I think you may be right, this double-indexing version may not work even in numpy.
So this funky way of indexing was done specifically to deal with tensor format, but the original numpy has a single continuous indexing block i.e. [index1,index2+index3]…
I think I need to figure out how to do that in pytorch, if that’s possible at all!?
EDIT: Thanks for pointing this out!!