multik-core
0.2.0
common
multik-core
/
org.jetbrains.kotlinx.multik.api.math
/
Math
/
argMaxDN
arg
Max
DN
common
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
.