ENH: torch: allow negative indices in take and take_along_axis
#1381
| Job | Run time |
|---|---|
| 43s | |
| 42s | |
| 42s | |
| 1m 12s | |
| 58s | |
| 1m 3s | |
| 1m 6s | |
| 6m 26s |
take and take_along_axis
#1381
| Job | Run time |
|---|---|
| 43s | |
| 42s | |
| 42s | |
| 1m 12s | |
| 58s | |
| 1m 3s | |
| 1m 6s | |
| 6m 26s |