torch.Tensor.argmax =================== .. currentmodule:: torch .. automethod:: Tensor.argmax