view()和squeeze()函数区别

最新推荐文章于 2023-09-15 16:18:58 发布

杨菲菲ya

最新推荐文章于 2023-09-15 16:18:58 发布

阅读量1k

点赞数

本文链接：https://blog.csdn.net/feifeiyaa/article/details/103929687

版权

最近调程序的时候，需要将三维变为二维，想到使用==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]