增加一个维度 out.unsqueeze(-1)
降低一个维度 out.squeeze(dim=1)
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:pytorch 矩阵数据增加维度unsqueeze和降低维度squeeze - Python技术站
增加一个维度 out.unsqueeze(-1)
降低一个维度 out.squeeze(dim=1)
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:pytorch 矩阵数据增加维度unsqueeze和降低维度squeeze - Python技术站