i have a tensor, A, of shape (155,1,79,95,68)
i have a tensor, Aind of shape (6200,5) containing indices of certain elements in A.
i want to do something like A[Aind] and receive a tensor of shape (6200,1) or vice versa, with the elements corresponding to those indices.
when i actually do so, i get not what i’d expect.
so far, the very clumsy solution i found is:
what_i_want = A[Aind[:,0],Aind[:,1],Aind[:,2],Aind[:,3],Aind[:,4]]