torch.max-JZTXT

dim (可选): 沿着哪个维度计算最大值，默认是计算整个张量的最大值

写法1
x = torch.tensor([[1, 2, 3],
                  [4, 5, 6]])

values,_= torch.max(x, dim=0)
print(values)   # 输出每行的最大值 [3, 6]
tensor([4, 5, 6])

写法2

x = torch.tensor([[1, 2, 3],
                  [4, 5, 6]])

value= torch.max(x, dim=0).values
print(value)