pytorch学习记录2--tensor维度变化

1. view / reshape  #改变维度为指定维度

• a=torch.rand(2,3,28,28)
• a.view(2,-1) .shape
• a.reshape(2,-1).shape

输出结果为：

torch.Size([2, 2352])

torch.Size([2, 2352])

 

2.squeeze / unsqueeze #压缩或扩展维度

squeeze用于维度压缩

• b=torch.rand(1,32,1,1)
• print(b.squeeze(0).shape) #若当前索引shape！=1，则不会变

输出结果为：

torch.Size([32, 1, 1])

unsqueeze用于维度扩张

• a=torch.rand(2,3,28,28)
• print(a.unSqueeze(0).shape) #在0上增加一个维度

输出结果：

torch.