torch比较操作 包含torch.max()

torch.max()函数表达式

output = torch.max(input, dim)

输入

• input是输入
• dim是max函数索引的维度0/1，0是每列的最大值，1是每行的最大值

输出

• 函数会返回两个tensor，第一个tensor是每行的最大值；第二个tensor是每行最大值的索引。

>>> a = torch.randn(4, 4)
>>> a
tensor([[ 0.4067, -0.7722, -0.6560, -0.9621],
        [-0.8754,  0.0282, -0.7947, -0.1870],
        [ 0.4300,  0.5444,  0.3180,  1.2647],
        [ 0.0775,  0.5886,  0.1662,  0.8986]])
>>> torch.max(a, 1)
torch.return_types.max(
values=tensor([0.4067, 0.0282, 1.2647, 0.8986]),
indices=tensor([0, 1, 3, 3]))

实际上，torch.max()这个函数返回的是两个值，第一个值是具体的values（我们用下划线_表示），第二个值是value所在的indices（也就是predicted）。

_, predicted = torch.max(outputs.data, 1)

因为我们不关心最大值是什么，而关心最大值对应的index是什么，所以选用下划线代表不需要用到的变量。

torch的其他比较操作