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

.. currentmodule:: jax.lax

.. autofunction:: dynamic_update_index_in_dim