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