美文网首页
pytorch中的squeeze和unsqueeze

pytorch中的squeeze和unsqueeze

作者: zhnidj | 来源:发表于2018-08-19 23:07 被阅读0次

十分有用,前者用来缩减维度,比如从(64,64,1,3,3)到(64,64,3,3),后者用来增加维度,比如从(64,64,3,3)到(64,64,1,3,3),自定义加载参数方式时候以及其他一些时候会用到

相关文章

网友评论

      本文标题:pytorch中的squeeze和unsqueeze

      本文链接:https://www.haomeiwen.com/subject/isbhiftx.html