optax.tree_utils.tree_zeros_like#
- optax.tree_utils.tree_zeros_like(
- tree: Any,
- dtype: str | type[Any] | dtype | SupportsDType | None = None,
Creates an all-zeros tree with the same structure.
- Parameters:
tree – pytree.
dtype – optional dtype to use for the tree of zeros.
- Returns:
an all-zeros tree with the same structure as
tree.