【学习笔记】pytorch中squeeze()和unsqueeze()函数介绍

squeeze用来减少维度， unsqueeze用来增加维度
具体可见下方博客。
pytorch中squeeze和unsqueeze

posted @ 2021-01-18 11:25 Curryxin 阅读(2079) 评论(0) 收藏举报

刷新页面返回顶部

Live2D