fun <T : Number> argMax(a: KtNDArray<T>): Long (source)
Returns the index of the maximum value along the flattened array.
Return
index of type Long.
See Also
fun <T : Number> argMax(a: KtNDArray<T>, axis: Int): KtNDArray<Long> (source)
Returns the indices of the maximum values along an axis.
axis - indices along the specified axis.
Return
KtNDArray of indices into the array. It has the same shape as a with the dimension along axis removed.
See Also