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)