会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
zae
首页
新随笔
管理
2021年11月14日
pytorch的argmax:只改变要挑选的维度
摘要: 只改变要挑选的维度,其他维度不变 A=torch.tensor([[[3,4]]]) dec_X = A.argmax(dim=2)# 只在dim=2上挑选最大值,得到索引为scalar dec_X #相当于把最内侧的[3,4]的维度去掉了,得到结果1。其他维度不变 tensor([[1]])
阅读全文
posted @ 2021-11-14 17:36 zae
阅读(139)
评论(0)
推荐(0)
编辑