pytorch中改变张量形状:torch.view()、torch.squeeze() / torch.unsqueeze()、torch.permute()
torch.view()
- view函数使用的前提是改变前后数字的总个数不能变,(4,4)能变成(2,8),但是不能变成(3,7)
- 而且view是不会原地更改的,比如下面这样,a还是原来的a,得把view之后数的存到新的变量去
- view里也可以填-1,函数会自己计算应该变成什么样子
- 但是只能有一个-1,多了要报错
- 而且也不能让计算结果变成一个除不尽的数,比如这个4×4×4×4=256,除以3×8=24,256/24除不尽,就会报错