Hi,
I am finding a weird behavior using tensors for indexing, so wanted some insight.
x=torch.tensor([[1,2],[3,4],[5,6]])
Now,
x[torch.tensor([[0],[1]])]
tensor([[[ 1,  2]],
        [[ 3,  4]]])

but what I expect is as
x[[[0],[1]]]
tensor([ 2])

How can I get this result using tensor indexing. Thank you.