torch:argmax

(torch:argmax a &key axis)

Non-differentiable: returns the index of the largest element (linalg:argmax) as a raw value, not a tensor -- the integer index for a vector, the per-slice index array with :axis. The indices feed torch:gather / torch:index-select, and greedy decoding reads its result directly.