代码:
a = torch.tensor([[4, 2, 3],
[1, 6, 5]])
print(a.size())
print('==========')
print(a.max(0))
print('==========')
print(a.max(1))
输出:
torch.Size([2, 3])
==========
torch.return_types.max(
values=tensor([4, 6, 5]),
indices=tensor([0, 1, 1]))
==========
torch.return_types.max(
values=tensor([4, 6]),
indices=tensor([0, 1]))
分析:
max返回值有两个,一个是最大值,另一个是坐标。对于二维张量来说,max(0)表示返回每列的最大值以及其列坐标,max(1)表示返回每行的最大值及其行坐标