jax.numpy.take\_along\_axis
===========================

.. currentmodule:: jax.numpy

.. autofunction:: take_along_axis