如何逆转张量中行的顺序

Question

我试图在一个张量中颠倒行的顺序，这是我创建的。我试过用tensorflow和Py手电筒。我发现的唯一东西是torch.flip()方法。这不起作用，因为它不仅颠倒了行的顺序，而且逆转了每一行中的所有元素。我希望元素保持不变。是否有数组操作来索引整数？例如：

 tensor_a = [1, 2, 3]
       [4, 5, 6]
       [7, 8, 9]

I want it to be returned as:
       [7, 8, 9]
       [4, 5, 6]
       [1, 2, 3]

 however, torch.flip(tensor_a) = 
       [9, 8, 7]
       [6, 5, 4]
       [3, 2, 1]

有人有什么建议吗？

Answer 1

根据文档，torch.flip有参数dims，它控制要翻转的轴。在这种情况下，torch.flip(tensor_a, dims=(0,))将返回预期的结果。此外，torch.flip(tensor_a)将反转所有张量，而torch.flip(tensor_a, dims=(1,))将反转每一行，如[1, 2, 3] --> [3, 2, 1]。