argmax

import { argmax } from "destack:tensor";

Return indices of maximum values along axis.

export declare function argmax<T, const Rank: usize>(tensor: Tensor<T, Rank>, axis: Axis, options?: TensorOutputOptions<usize, Rank - 1> | undefined): Tensor<usize, Rank - 1>
language/library/src/tensor/ops.ds:355:359