本文主要是介绍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列。
太容易忘记，留下记录，非常基础的知识，年龄太大，肤浅得不好意思
这篇关于torch.max()学习记录的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
