optax.losses.cosine\_similarity
===============================

.. currentmodule:: optax.losses

.. autofunction:: cosine_similarity