kmath
0.3.1-dev-RC
jvm
kmath-nd4j
/
space.kscience.kmath.nd4j
/
Nd4jTensorAlgebra
/
argMax
arg
Max
open
override
fun
StructureND
<
T
>
.
argMax
(
dim
:
Int
,
keepDim
:
Boolean
)
:
Tensor
<
Int
>
(
source
)