ENH: torch: allow negative indices in take and take_along_axis
#1125
| Job | Run time |
|---|---|
| 3m 8s | |
| 3m 2s | |
| 6m 10s |
take and take_along_axis
#1125
| Job | Run time |
|---|---|
| 3m 8s | |
| 3m 2s | |
| 6m 10s |