十分有用,前者用来缩减维度,比如从(64,64,1,3,3)到(64,64,3,3),后者用来增加维度,比如从(64,64,3,3)到(64,64,1,3,3),自定义加载参数方式时候以及其他一些时候会用到
十分有用,前者用来缩减维度,比如从(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
网友评论