ENH: torch: allow negative indices in take and take_along_axis
#1356
| Job | Run time |
|---|---|
| 4m 40s | |
| 4m 20s | |
| 4m 36s | |
| 13m 36s |
take and take_along_axis
#1356
| Job | Run time |
|---|---|
| 4m 40s | |
| 4m 20s | |
| 4m 36s | |
| 13m 36s |