ENH: torch: allow negative indices in take and take_along_axis
#846
| Job | Run time |
|---|---|
| 3m 20s | |
| 3m 34s | |
| 3m 9s | |
| 10m 3s |
take and take_along_axis
#846
| Job | Run time |
|---|---|
| 3m 20s | |
| 3m 34s | |
| 3m 9s | |
| 10m 3s |