optax.tree_utils.tree_div

Contents

optax.tree_utils.tree_div#

optax.tree_utils.tree_div(tree_x: Any, tree_y: Any) Any[source]#

Divide two pytrees.

Parameters:
  • tree_x โ€“ first pytree.

  • tree_y โ€“ second pytree.

Returns:

the quotient of the two pytrees.