会员
周边
众包
新闻
博问
闪存
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
阿洛萌萌哒
博客园
首页
新随笔
联系
管理
订阅
2020年10月22日
PyTorch-->torch.max()的用法
摘要: _, predited = torch.max(outputs,1) # 此处表示返回一个元组中有两个值,但是对第一个不感兴趣 返回的元组的第一个元素是image data,即是最大的值;第二个元素是label,即是最大的值对应的索引。由于我们只需要label(最大值的索引),所以有 _ , pre
阅读全文
posted @ 2020-10-22 18:49 阿洛萌萌哒
阅读(2307)
评论(0)
推荐(0)
编辑
公告