会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Mr.Higgerw
越努力,越幸运
博客园
首页
联系
管理
2020年11月26日
pytorch的unsqueeze与squeeze函数
摘要: torch.squeeze() 这个函数主要对数据的维度进行压缩,去掉维数为1的的维度。 比如是一行或者一列这种,一个一行三列(1,3)的数去掉第一个维数为一的维度之后就变成(3)行。 squeeze(a)就是将a中所有为1的维度删掉。不为1的维度没有影响。 a.squeeze(dim) 就是去掉a
阅读全文
posted @ 2020-11-26 11:30 爱吃砂糖橘的白龙
阅读(242)
评论(0)
推荐(0)
编辑
公告