点此进入CSDN

点此添加QQ好友 加载失败时会显示




你的浏览器不支持播放哦!!nuttertools 您的浏览器不支持该播放!

pytorch 矩阵数据增加维度unsqueeze和降低维度squeeze

增加一个维度 out.unsqueeze(-1)

降低一个维度 out.squeeze(dim=1)

 

posted @ 2020-02-23 11:32  高颜值的殺生丸  阅读(1338)  评论(0)    收藏  举报

作者信息

昵称:

刘新宇

园龄:4年6个月


粉丝:1209


QQ:522414928