squeeze和unsqueeze作用

作用

改变张量的维度，便于进行运算。如果你当前的张量shape为三维，而需要四维才能参与运算，这时候就可以添加一个维度，维数为1。

示例

我们首先生成一个二维张量 b，接着生成三维张量 c，四维张量d，再缩回二维张量 e。


从例子中可以发现：

1. unsqueeze(xxx) 可以在第 xxx 维插入一个维度，维数为 1 。
2. squeeze() 可以把张量所有的维数为 1 的维度去掉。