pytorch 去除维度为1的维度 out.squeeze(dim=1) out.squeeze_(dim=1) 多思考也是一种努力，做出正确的分析和选择，因为我们的时间和精力都有限，所以把时间花在更有价值的地方。