optax.tree_utils.tree_sub# optax.tree_utils.tree_sub(tree_x: Any, tree_y: Any) → Any[source]# Subtract two pytrees. Parameters: tree_x – first pytree. tree_y – second pytree. Returns: the difference of the two pytrees.
