去掉张量里的无用维度

https://pytorch.org/docs/stable/generated/torch.squeeze.html#torch.squeeze
TORCH.SQUEEZE

对于一个tensor

如果他的shape是

3 $ \times $ 7 $ \times $ 1 $ \times $ 4 $ \times $ 1

可以用

torch.squeeze(input, dim=None, *, out=None) → Tensor

去掉没用的1，变成3 $ \times $ 7 $ \times $ 4

因为这里的1是没有意义的，比如3 $ \times $ 1的矩阵其实这里的3是有数据的，但是1是没有数据的，3 $ \times $ 2开始，第二个维度才有实际的数据，所以去掉1对张量本身不会造成任何影响