torch.Tensor.argmax

Tensor.argmax(dim=None, keepdim=False) → LongTensor

参见 torch.argmax()