pytorch tensor 调换矩阵行的顺序

生成一个想要的顺序 [1, 0] 然后去索引