view()和squeeze()函数区别

最近调程序的时候，需要将三维变为二维，想到使用==squeeze()==这个函数，在使用的时候，发现squeeze（）它只能对维数中含有1的降维，例如：

X = torch.rand(40, 10,128)        # size:[40,10,128]Y = torch.rand(1, 10,128)         # size:[1,10,128]X = X.squeeze(0)                  # size:[40,10,128]Y = Y.squeeze(0)                  # size:[10,128]

但是view()函数可以对尺寸进行调整，若view()函数中的参数含有-1,这表示根据其他尺寸随机调整这个维度的维数，例如：

X = torch.rand(40, 10,128)X= X.view(-1,400,128)      # size:[1,400.128]Z= X.view(-1,40,128)       # size:[10,40.128]Y = X.view(400,128)     # size:[400.128]X = X.squeeze()          # size:[400.128]