optax.tree\_utils.tree\_where
=============================

.. currentmodule:: optax.tree_utils

.. autofunction:: tree_where