jax.numpy.take ============== .. currentmodule:: jax.numpy .. autofunction:: take