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