torch.Tensor.max

Tensor.max(dim=None, keepdim=False)

请参阅 torch.max()