评价此页

torch.Tensor.argmax#

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

请参阅 torch.argmax()