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.
