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

argMaxD4

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

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