torch.max()学习记录

x=tourch.tensor([[1,2,3],[4,6,5],[9,11,4],[-2,6,20]])为4*3的一个张量

1)请思考y的值

y=torch.max(x,dim=0)  

分析：x是一个2D: 4*3的张量，dim=0 ，表示按照行操作，得到的张量shape=3,对应的是[1,4,9,-2】，【2,6,11,6】，【3,5,4,20】三组内挑选最大值组分别是9、11、20成为【9,11,20】

因此最大应该是y=tensor([9,11,20]) ,对应的argmax()=tensor([2,2,3]),表示【9,11,20】三个分别在第2行、第二行、第三行。

2)请思考y的值

y=torch.max(x,dim=1)  

分析：x是一个2D: 4*3的张量，dim=1 ，表示按照列操作，得到的张量shape=4,我的理解是[1,2,3]的最大为3，[4,5,6]的最大为6，【9,11,4]最大值为11，,[-2,6,20]最大值为20，因此y=tensor([3,6,11,20]),对应的argmax()=tensor([2,1，1,2]),表示【3,6,11,20】中，3处于第二列，6处于第1列,11处于第1列，20处于第2列。

太容易忘记，留下记录，非常基础的知识，年龄太大，肤浅得不好意思