argMax

open override fun Tensor<T>.argMax(dim: Int, keepDim: Boolean): MultikTensor<T>

Sources

jvm source
Link copied to clipboard