ENH: torch: allow negative indices in take and take_along_axis
#1395
| Job | Run time |
|---|---|
| 1m 0s | |
| 1m 9s | |
| 39s | |
| 46s | |
| 1m 8s | |
| 39s | |
| 1m 13s | |
| 6m 34s |
take and take_along_axis
#1395
| Job | Run time |
|---|---|
| 1m 0s | |
| 1m 9s | |
| 39s | |
| 46s | |
| 1m 8s | |
| 39s | |
| 1m 13s | |
| 6m 34s |