jax.lax.iota
============

.. currentmodule:: jax.lax

.. autofunction:: iota