pytorch中的torch.squeeze()函数

torch.squeeze(input, dim=None, out=None)

squeeze()函数的功能是维度压缩。返回一个tensor（张量），其中 input 中大小为1的所有维都已删除。

如果 input 的形状为 (A×1×B×C×1×D)，那么返回的tensor的形状则为 (A×B×C×D)

维度从0开始计数。

设定 dim 参数，只在给定的维度（dimension）上进行压缩操作。

如果 input 的形状为 (A×1×B)，squeeze(input, 0)后，返回的tensor不变；squeeze(input, 1)后，返回的tensor将被压缩为 (A×B)

上次tensor（[16,1,1,1]),需要转换成tensor（[16,1]),，就写 torch.squeeze(input, dim=2，3)，转换成功。