optax.tree\_utils.tree\_scale
=============================

.. currentmodule:: optax.tree_utils

.. autofunction:: tree_scale