squeeze 、unsequeeze

两者都是作用于tensor。

记住单词：squeeze是挤压、塞入的意思

unsqueeze(num)是增加维度，num代表增加维度所在的位置。

squeeze(num)是减少维度，num代表的是减少维度所在的位置，所减少的维度只能是1维，也就是不存在数据的维度。

import torch
a=torch.arange(0,6)
a=a.view(2,3)
print(a.shape)#torch.Size([2,3])

b=a.unsqueeze(1)
print(b.shape)#torch.Size([2,1,3])
c=a.squeeze(0)
print(c.shape)#torch.Size([2,3])
d=a.unsqueeze(2)
print(d.shape)#torch.Size([2,3,1])
e=d.squeeze(2)
print(e.shape)#torch.Size([2,3])

version.shape可查看tensor的shape。