Hi, I have two different machines both Ubuntu 18.04 with torch 1.9.0 and 1.9.0+cu102. When I try the followings:
V = torch.tensor([[1,2,3],[4,5,6]])
idx = torch.tensor([0])

Now if I try:
print(V[idx]) 

I ll get different outputs:
tensor([[1, 2, 3]]) # for torch 1.9.0
tensor([1, 2, 3]) # for torch 1.9.0+cu102

Do you have any idea?