flax.core.frozen_dict package
flax.core.frozen_dict package#
- class flax.core.frozen_dict.FrozenDict(*args, __unsafe_skip_copy__=False, **kwargs)[source]#
An immutable variant of the Python dict.
- pop(key)[source]#
Create a new FrozenDict where one entry is removed.
Example:
state, params = variables.pop('params')
- Parameters
key – the key to remove from the dict
- Returns
A pair with the new FrozenDict and the removed value.