去掉张量里的无用维度

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对张量本身不会造成任何影响