jax
fc170b2b - Merge pull request #236 from google/numpy-take-along-axis

Commit
6 years ago
Merge pull request #236 from google/numpy-take-along-axis Add numpy.take_along_axis (fixes #220)
Author
Loading