flax.core.frozen_dict 包

class flax.core.frozen_dict.FrozenDict(*args, __unsafe_skip_copy__=False, **kwargs)

Python 字典的不可变变体。

copy(add_or_replace=mappingproxy({}))

创建一个带有附加或替换条目的新 FrozenDict。

pop(key)

创建一个删除一个条目的新 FrozenDict。

示例

>>> from flax.core import FrozenDict
>>> variables = FrozenDict({'params': {...}, 'batch_stats': {...}})
>>> new_variables, params = variables.pop('params')

参数

key – 要从字典中删除的键

返回

一对包含新的 FrozenDict 和删除的值。

pretty_repr(num_spaces=4)

返回嵌套字典的缩进表示形式。

unfreeze()

解冻此 FrozenDict。

返回

此 FrozenDict 实例的未冻结版本。

flax.core.frozen_dict.freeze(xs)

冻结嵌套字典。

通过将其转换为 FrozenDict 使嵌套的 dict 不可变。

参数

xs – 要冻结的字典（一个常规 Python 字典）。

返回

冻结的字典。

flax.core.frozen_dict.unfreeze(x)

解冻 FrozenDict。

通过将其转换为（嵌套）字典，使 FrozenDict 的可变副本变为可变。

参数

x – 要解冻的冻结字典。

返回

未冻结的字典（一个常规 Python 字典）。

flax.core.frozen_dict.copy(x, add_or_replace=FrozenDict({}))

创建一个带有附加和/或替换条目的新字典。这是一个实用函数，可以对 FrozenDict 或常规字典起作用，并模仿 FrozenDict.copy 的行为。

示例

>>> from flax.core import FrozenDict, copy
>>> variables = FrozenDict({'params': {...}, 'batch_stats': {...}})
>>> new_variables = copy(variables, {'additional_entries': 1})

参数

• x – 要复制和更新的字典

• add_or_replace – 要添加到字典 x 或在字典 x 中替换的键值对的字典

返回

一个带有附加和/或替换条目的新字典。

flax.core.frozen_dict.pop(x, key)

创建一个删除一个条目的新字典。这是一个实用函数，可以对 FrozenDict 或常规字典起作用，并模仿 FrozenDict.pop 的行为。

示例

>>> from flax.core import FrozenDict, pop
>>> variables = FrozenDict({'params': {...}, 'batch_stats': {...}})
>>> new_variables, params = pop(variables, 'params')

参数

• x – 要从中删除条目的字典

• key – 要从字典中删除的键

返回

一对包含新字典和删除的值。

flax.core.frozen_dict.pretty_repr(x, num_spaces=4)

返回嵌套字典的缩进表示形式。这是一个实用函数，可以对 FrozenDict 或常规字典起作用，并模仿 FrozenDict.pretty_repr 的行为。如果 x 是任何其他 dtype，则此函数将返回 repr(x)。

参数

• x – 要表示的字典

• num_spaces – 每个缩进级别的空格字符数

返回

嵌套字典的缩进字符串表示形式。