jax.lax.axis\_size
==================

.. currentmodule:: jax.lax

.. autofunction:: axis_size