5_torch.max的用法举例

1. 求一维的Array的max

x = torch.tensor([1, 5, 3, 9, 2])
max_value = torch.max(x)
print(max_value
1
2