首页 > 其他 > 详细

torch.max和torch.argmax

时间:2021-07-28 18:34:22      阅读:9      评论:0      收藏:0      [点我收藏+]

结论
torch.max(),不指定dim参数,返回输入张量中所有数据的最大值; 如果指定dim参数,则返回按照指定维度的最大值和各最大值对应的位置
torch.argmax(),不指定dim,然后输入张量中所有数据最大值位置(按照张量被拉伸为1为向量算); 如果指定dim,返回指定维度的最大值位置,另外可以通过keepdim来保留原张量的形状

torch.max和torch.argmax

原文:https://www.cnblogs.com/pyclq/p/15071494.html

(0)
(0)
   
举报
评论 一句话评论(0
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!