squeeze的用法主要就是对数据的维度进行压缩或者解压。
squeeze() torch.squeeze(a):去掉a中维数为1的维度。 a.squeeze(N):去掉特定维度N下维数为1的维度。 b=torch.squeeze(a,N) :a中去掉指定的维数为1的维度。
torch.squeeze(a)
a.squeeze(N)
b=torch.squeeze(a,N)
unsqueeze() a.unsqueeze(N) :在a中指定位置N加上一个维数为1的维度。 b=torch.unsqueeze(a,N):在a中指定位置N加上一个维数为1的维度。
a.unsqueeze(N)
b=torch.unsqueeze(a,N)