如何在张量中反转行的顺序

Question

我正在尝试反转我创建的张量中的行的顺序。我试过tensorflow和pytorch。我只找到了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]一样。