netket.jax.tree_to_real

netket.jax.tree_to_real(pytree)

Replace all complex leaves of a pytree with a RealImagTuple of 2 real leaves.

Parameters

pytree (Any) – a pytree to convert to real

Return type

Tuple[Any, Callable]

Returns

A pair where the first element is the converted real pytree, and the second element is a callable for converting back a real pytree to a complex pytree of of the same structure as the input pytree.