torch中的squeeze和unsqueeze
squeeze是压缩,对维度进行降维。不写的话,默认将所有维度为1的去掉(我理解就是去掉对应层的"[]"中括号)
举例:
unsqueeze是和squeeze相反的操作
标签:unsqueeze,记录,维度,实习,去掉,squeeze From: https://www.cnblogs.com/Halo-zyh-Go/p/17526130.html
torch中的squeeze和unsqueeze
squeeze是压缩,对维度进行降维。不写的话,默认将所有维度为1的去掉(我理解就是去掉对应层的"[]"中括号)
举例:
unsqueeze是和squeeze相反的操作
标签:unsqueeze,记录,维度,实习,去掉,squeeze From: https://www.cnblogs.com/Halo-zyh-Go/p/17526130.html