i think you keep forgetting pytorch derived from torch, not numpy.
we fixed as much differences as we can, but torch always returned indices in torch

→ View original post on X — @soumithchintala