pytorch ：使用两次sort函数（排序）找出矩阵每个元素在升序或降序排列中的位置

对于一个二维矩阵，使用两次sort函数（对原矩阵、第一次sort之后的index）后，index（第二个变量）的输出是原矩阵一个由大到小（或由小到大）的映射，映射的值为整数，位置保持不变。