pytorch中max的用法

对pytorch中的数据使用max()函数的时候，结果是含有2个元素元组，第一个元素是对应计算维度最大值的tensor，第二个元素是对应的计算维度最大值的索引。需要注意，对应计算维度就是把哪个维度去掉，如维度为（4,5），如果max(0)，则结果的维度为5列，其中每一列为原每一列的最大值。如维度为（3,4,5），则max(0)，max(1)，max(2)，分别对应的输出维度为（4,5），（3,5），（3,4）。

示例如下：

a = torch.randn(10)
print(a)
tensor([-0.6255, -0.9773,  1.0860, -0.7099, -0.1986, -0.2131,  2.3024, -1.0392,
        -1.0161, -0.1241])
b
b = a.view(-1, 5)
print(b)
tensor([[-0.6255, -0.9773,  1.0860, -0.7099, -0.1986],
        [-0.2131,  2.3024, -1.0392, -1.0161, -0.1241]])
print(b.max(0))
print(b.max(0)[0])
print(b.max(0)[1])
print(b.max(1))
print(b.max(1)[0])
print(b.max(1)[1])
torch.return_types.max(
values=tensor([-0.2131,  2.3024,  1.0860, -0.7099, -0.1241]),
indices=tensor([1, 1, 0, 0, 1]))
tensor([-0.2131,  2.3024,  1.0860, -0.7099, -0.1241])
tensor([1, 1, 0, 0, 1])
torch.return_types.max(
values=tensor([1.0860, 2.3024]),
indices=tensor([2, 1]))
tensor([1.0860, 2.3024])
tensor([2, 1])

详细解释可参考如下链接：

pytorch学习——torch.max() 和 torch.gather()_岛崎三步的博客-CSDN博客