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

argMaxDN

abstract fun <T : Number> argMaxDN(a: MultiArray<T, DN>, axis: Int): NDArray<Int, D4>
Content copied to clipboard

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