PyTorch基本操作

 1.改变tensor形状 a.view(row,column)/a.reshape(row,column)

import torch
a=torch.rand(4,3)
a.view(2,6)
a.shape ##(2,6)
##a.reshape(2.6)##可以有相同效果

2. 扩展tensor

import torch
a=torch.rand(2,3)
print(a)
r=a.unsqueeze(1)##在axis=1的加入一个维度 且为1
print(r)
r.shape ##(2,1,3)

3.压缩tensor

c=r.squeeze(1) ##看r.shape中axis=1的值是否为1，是的话可以去除该维度，不是的话该指令无效
c.shape #（2,3)
b=r.squeeze(0) ##看r.shape中axis=0的值是否为1，是的话可以去除，不是的话该指令无效
b.shape # (2,1,3)