Jupyter Notebook 查找矩阵行列式的根时出现弃用警告

Question

zhixin

Asked: 2023-12-20 21:49:58 +0800 CST2023-12-20 21:49:58 +0800 CST 2023-12-20 21:49:58 +0800 CST

如何按特定键值“批量”排序张量？

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

原始批量矩阵（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 循环解决问题，这不是并行的。那么如何并行处理整个批次呢？

Salvatore Daniele Bianco · Answer 1 · 2023-12-20T22:31:10+08:00

Best Answer

Salvatore Daniele Bianco

结果可能有点令人困惑，但很有意义：

(my_tensor[:,torch.argsort(my_tensor[:,:,0], dim=1)])\
[torch.arange(len(my_tensor)),torch.arange(len(my_tensor))]

在第一行中，您提取排序张量思想torch.argsort并将其应用于my_tensor，从而产生(2, 2, 3, 2)形状张量。由于您希望每个元素仅根据其第一列进行排序，因此您只对前两个维度的对角线感兴趣，并且可以通过切片（第二行代码）来提取它。