chore: use xp.take_along_axis
if Array API version >=2024.12
#6025
Job | Run time |
---|---|
26s | |
26s |
xp.take_along_axis
if Array API version >=2024.12
#6025
Job | Run time |
---|---|
26s | |
26s |