如果只输入一个张量作为参数,那么不管输入的张量是几维的,返回的都是一个数字张量,它是输入张量的最大值。
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)