optax.tree_utils.tree_conj

Contents

optax.tree_utils.tree_conj#

optax.tree_utils.tree_conj(tree: Any) Any[source]#

Compute the conjugate of a pytree.

Parameters:

tree โ€“ pytree.

Returns:

a pytree with the same structure as tree.