Pytorch | torch.max（）的用法

如果只输入一个张量作为参数，那么不管输入的张量是几维的，返回的都是一个数字张量，它是输入张量的最大值。

xx=torch.tensor([1,2,3,4])
print(torch.max(xx))
yy=torch.tensor([[1,2,3,4],[2,5,1,3]])
print(torch.max(yy))

如果输入是多维张量，可以多一个"维度”参数。例如，max(tensor1,0)表示想在第0维找最大值，max(tensor1,1)表示想在第1维找最大值。这时就会返回两个tensor，分别表示值和位置。

xx=torch.tensor([[1,2,3,4],[2,5,1,3]])
yy=torch.max(xx,0)
print(yy)
print("\n")
i,j=yy
print(i,j)