optax.tree_utils.tree_size# optax.tree_utils.tree_size(tree: Any) → int[source]# Total size of a pytree. Parameters: tree – pytree Returns: the total size of the pytree.
