增加一个维度 out.unsqueeze(-1)
降低一个维度 out.squeeze(dim=1)
pytorch 矩阵数据增加维度unsqueeze和降低维度squeeze
原文：https://www.cnblogs.com/LiuXinyu12378/p/12348907.html
