[ Pytorch ] torch.squeeze() 和torch.unsqueeze()的用法

squeeze的用法主要就是对数据的维度进行压缩或者解压。

squeeze()
torch.squeeze(a)：去掉a中维数为1的维度。
a.squeeze(N)：去掉特定维度N下维数为1的维度。
b=torch.squeeze(a，N) ：a中去掉指定的维数为1的维度。

unsqueeze()
a.unsqueeze(N) ：在a中指定位置N加上一个维数为1的维度。
b=torch.unsqueeze(a，N)：在a中指定位置N加上一个维数为1的维度。