kmath-tensorflow/space.kscience.kmath.tensorflow/TensorFlowAlgebra/argMax

argMax

open override fun StructureND<T>.argMax(dim: Int, keepDim: Boolean): Tensor<Int>
Content copied to clipboard

Sources

Link copied to clipboard