torch.max和torch.argmax

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

torch.max和torch.argmax

上一篇:element ui框架中的el-scrollbar组件滚动条监测


下一篇:wx.showModal() 内容如何换行?