点此进入CSDN

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




pytorch 去除维度为1的维度

 

out.squeeze(dim=1)

 

out.squeeze_(dim=1)

posted @ 2020-02-20 18:14  高颜值的殺生丸  阅读(1263)  评论(0)  编辑  收藏  举报

作者信息

昵称：

刘新宇

园龄：4年6个月


粉丝：1209


QQ：522414928