chore: use `xp.take_along_axis` if Array API version >=2024.12#4406

Merged

wanghan-iapcm merged 1 commit intodeepmodeling:develfrom

njzjz:xp-take-along-axis

Nov 23, 2024

Commits on Nov 22, 2024

chore: use `xp.take_along_axis` is Array API version >=2024.12
njzjz
committed