pytorch 矩阵数据增加维度unsqueeze和降低维度squeeze

增加一个维度 out.unsqueeze(-1)

降低一个维度 out.squeeze(dim=1)