将torch.tensor([[1,2,3],[4,5]]）实现为torch.tensor([[3,2,1],[5,4]]）

在PyTorch中，如果你想要将tensor `[[1, 2, 3], [4, 5]]` 的元素顺序反转，你可以直接使用切片操作或者numpy的 transpose 功能。这里提供两种方法：

方法一：使用numpy的transpose功能

import torch
import numpy as np

# 创建原始 tensor
input_tensor = torch.tensor([[1, 2, 3], [4, 5]])
# 转换为numpy数组并反转轴
numpy_array = input_tensor.numpy().T
# 再转换回torch tensor
reversed_tensor = torch.from_numpy(numpy_array)
print(reversed_tensor)

方法二：直接在torch上进行切片操作

reversed_tensor = input_tensor[:, ::-1]
print(reversed_tensor)

在这两个例子中，最后的 `reversed_tensor` 都会输出 `[[3, 2, 1], [5, 4]]`。