multik-core/org.jetbrains.kotlinx.multik.api.math/Math/argMaxD2

argMaxD2

abstract fun <T : Number> argMaxD2(a: MultiArray<T, D2>, axis: Int): NDArray<Int, D1>
Content copied to clipboard

Returns an ndarray of indices of maximum elements in a two-dimensional ndarray a over a given axis.