ENH: torch: allow negative indices in take and take_along_axis
#1122
| Job | Run time |
|---|---|
| 4m 37s | |
| 5m 14s | |
| 9m 51s |
take and take_along_axis
#1122
| Job | Run time |
|---|---|
| 4m 37s | |
| 5m 14s | |
| 9m 51s |