ENH: torch: allow negative indices in take and take_along_axis
#830
| Job | Run time |
|---|---|
| 4m 17s | |
| 3m 12s | |
| 7m 29s |
take and take_along_axis
#830
| Job | Run time |
|---|---|
| 4m 17s | |
| 3m 12s | |
| 7m 29s |