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