optax.init_empty_state

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 GradientTransformation with empty state.