argMax

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

Sources

Link copied to clipboard