PyTorch Squeeze and Unsqueeze

Squeeze : 挤出Dimension 的水分

Squeeze 会消除只有一个元素的维度，使得Tenson 的维度降下来:

x = torch.zeros(2, 1, 2, 1, 2)
x.size()

>>> torch.Size([2, 1, 2, 1, 2])

y = torch.squeeze(x)
y.size()

>>> torch.Size([2, 2, 2])

Dim 参数， 指定挤水分的位置:

如果对应位置维度是1， 那么就删除这个维度，否则就保存：

Dimension 0 的位置是2， 所以没有删除该维度

y = torch.squeeze(x, 0)
y.size()

>>> torch.Size([2, 1, 2, 1, 2])

Dimension 1 的位置是1， 所以删除该维度

y = torch.squeeze(x, 1)
y.size()

>>> torch.Size([2, 2, 1, 2])