freeze_state_vars#
- penzai.core.variables.freeze_state_vars(tree: Any, predicate: Callable[[StateVariable], bool] | None = None) Any[source]#
Version of
freeze_variablesthat only freezesStateVariables.
Version of freeze_variables that only freezes StateVariables.