小编Fer*_*nha的帖子

我有一个二维张量，我想像这个例子一样按第一维排序：

a = torch.FloatTensor(
   [[5, 5],
    [5, 3],
    [3, 5],
    [6, 4],
    [3, 7]])

排序后我期望这个结果：

a = torch.FloatTensor(
   [[3, 5],
    [3, 7],
    [5, 3],
    [5, 5],
    [6, 4]])

可以在pytorch中做到这一点吗？我知道可以在 numpy 中做到这一点，但我想使用火炬在 GPU 中做到这一点。