squeeze 删除张量维度中值为 1 的项，若原维度为 A*1*B*C*1*D，则删除后维度为 A*B*C*D x = torch.rand(size=(2, 1, 3)) print(x.shape) y = x.squeeze() print(y.shape) 输出结果： torch.Size