optax.init_empty_state#
- optax.init_empty_state(params: Array | ndarray | bool | number | bool | int | float | complex | Iterable[Array | ndarray | bool | number | bool | int | float | complex | Iterable[ArrayTree] | Mapping[Any, ArrayTree]] | Mapping[Any, Array | ndarray | bool | number | bool | int | float | complex | Iterable[ArrayTree] | Mapping[Any, ArrayTree]]) EmptyState[source]#
Init function for a
GradientTransformationwith empty state.