tensor笔记4--max()

代码： 

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)表示返回每行的最大值及其行坐标