argMax

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