pytorch系列：torch.squeeze

torch.squeeze(input, dim=None, out=None) 返回一个输入的所有规模为1的维移除的tensor。

如：
输入的shape为：$(A\times 1\times B\times C\times 1\times D)$，输出的tensor为：$(A\times B\times C\times D)$。

当给定$dim$时，squeeze操作只在给定维上进行，如果输入的shape为$(A\times 1\times B)$，那么squeeze(input, 0)则会保持原来的形状，但squeeze(input, 1)将会squeeze输入为$(A\times B)$的shape。

注意：返回的tensor和输入tensor是共享存储的，因此此处tensor内容的改变将会影响其他相应tensor的内容。