chenyuandGitHub 47aba47f64 update Torch.gather api (#4692) ...

* update Torch.gather api

gather(self, dim, index) to match torch

* fix that

2024-05-22 21:54:06 -04:00

..

__init__.py

more test cleanups (#2631)

2023-12-05 16:17:57 -08:00

test_indexing.py

update Torch.gather api (#4692)

2024-05-22 21:54:06 -04:00