netket.jax.tree_size

netket.jax.tree_size(tree)

Returns the sum of the size of all leaves in the tree. It’s equivalent to the number of scalars in the pytree.

Return type

int

Parameters

tree (Any) –