zhixin's questions

我需要按第一列的键值对一批二维矩阵的行进行排序：

原始批量矩阵（3d 张量）：

torch.tensor([[[2, 0], 
               [0, 1],
               [1, 2]],

              [[1, 2], 
               [0, 0], 
               [2, 1]]])

所需张量：

torch.tensor([[[0, 1],
               [1, 2],
               [2, 0]],

              [[0, 0],
               [1, 2],  
               [2, 1]]])

已经知道如何处理其中一个批次，另一个答案通过 for 循环解决问题，这不是并行的。那么如何并行处理整个批次呢？