jax.lax.dynamic\_index\_in\_dim
===============================

.. currentmodule:: jax.lax

.. autofunction:: dynamic_index_in_dim