I have the following tensor: vector = torch.tensor([[1,5,3], [2,3,4]])
vector = torch.tensor([[1,5,3], [2,3,4]])
How can I swap values on the second axis to have the following result: