optax.flatten
=============

.. currentmodule:: optax

.. autofunction:: flatten