pytorch 调整某一维度数据顺序的方法