optax.tree_utils.tree_sub

optax.tree_utils.tree_sub(

tree_x: Any,

tree_y: Any,

) → Any

Subtract two pytrees.

Parameters:

• tree_x – first pytree.

• tree_y – second pytree.

Returns:

the difference of the two pytrees.