optax.tree_utils.tree_mul

optax.tree_utils.tree_mul(

tree_x: Any,

tree_y: Any,

) → Any

Multiply two pytrees.

Parameters:

• tree_x – first pytree.

• tree_y – second pytree.

Returns:

the product of the two pytrees.