pytorch tensor 调换矩阵行的顺序

image

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

posted @ 2020-05-23 14:57  可爱小小畅  阅读(4040)  评论(0编辑  收藏  举报