arg
abstract fun <T : Number, D : Dimension> argMax(a: MultiArray<T, D>): Int
Content copied to clipboard
Returns flat index of maximum element in an ndarray.
abstract fun <T : Number, D : Dimension, O : Dimension> argMax(a: MultiArray<T, D>, axis: Int): NDArray<Int, O>
Content copied to clipboard
Returns an ndarray of indices of maximum elements in an ndarray a over a given axis.