optax.set\_to\_zero
===================

.. currentmodule:: optax

.. autofunction:: set_to_zero