torch:argmax

(torch:argmax a &key axis)

微分不可能な演算です。最大要素のインデックス (linalg:argmax) をテンソルではなく生の値として返します。ベクトルなら整数インデックス、:axis 付きならスライスごとのインデックス配列です。結果は torch:gather / torch:index-select に渡せ、貪欲デコードはこの値を直接読みます。