squeeze的用法主要就是对数据的维度进行压缩或者解压。
squeeze()
torch.squeeze(a)：去掉a中维数为1的维度。
a.squeeze(N)：去掉特定维度N下维数为1的维度。
b=torch.squeeze(a，N)：a中去掉指定的维数为1的维度。
unsqueeze()
a.squeeze(N)：在a中指定位置N加上一个维数为1的维度。
b=torch.squeeze(a，N)：在a中指定位置N加上一个维数为1的维度。
[ Pytorch ] torch.squeeze() 和torch.unsqueeze()的用法
原文：https://www.cnblogs.com/recoverableTi/p/12253515.html