the numpy compatible torch.compile behavior is definitely a bug. the expected behavior is the numpy output should be the same as torch.compile output.
torch's advanced indexing differing from numpy's advanced indexing is a bit of an aside.
I've just filed a bug report,
PyTorch torch.compile Advanced Indexing Bug Report Filed
By
–
Leave a Reply