1. torch.view()
data.view(a,b) # 把data 转换成 a*b 维的矩阵
data.view(k,-1)
2. torch.squeeze()
data.squeeze(0) # 表示在第一维度压缩,如果第一维度是1,则直接去掉
img=content
img.shape
torch.Size([1,3, 800, 600])
img=content.squeeze(0)
img.shape
torch.Size([3, 800, 600])
3. torch.unsqueeze()
与sequeeze() 相反,增加一个维度,维度为1