a = torch.tensor([1., 1., 1., 1., 0., 2.]).view(2, 3)
print(a)
print(a.flip(1))
------------------------
tensor([[1., 1., 1.],
        [1., 0., 2.]])

tensor([[1., 1.],
        [0., 1.]])

a = torch.tensor([1., 1., 1., 1., 0., 2.], device=torch.device('cuda')).view(2, 3)
print(a)
print(a.flip(1))
------------------------
tensor([[1., 1., 1.],
        [1., 0., 2.]], device='cuda:0')

tensor([[1., 1., 1.],
        [2., 0., 1.]], device='cuda:0')