unsqueeze 和 squeeze

squeeze压缩的意思 就是在第几维为1 去掉

unsqueeze 解缩 在第几维增加 变成*1

squeeze用法

c = b.view(1, 1, 1, 2, 3)
c.squeeze(0) # 压缩第0维的“１”
(0 ,0 ,.,.) =
  0  1  2
  3  4  5
[torch.FloatTensor of size 1x1x2x3]

c.squeeze() # 把所有维度为“1”的压缩
 0  1  2
 3  4  5
[torch.FloatTensor of size 2x3]

unsqueeze用法

b = a.view(-1, 3) # 当某一维为-1的时候，会自动计算它的大小
b
 0  1  2
 3  4  5
[torch.FloatTensor of size 2x3]

b.unsqueeze(1) # 注意形状，在第1维（下标从0开始）上增加“１”

(0 ,.,.) =
  0  1  2

(1 ,.,.) =
  3  4  5
[torch.FloatTensor of size 2x1x3]