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)

冻结一个嵌套字典。

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

参数

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

返回

冻结后的字典。

flax.core.frozen_dict.unfreeze(x)

解冻一个 FrozenDict。

通过将 FrozenDict 转换为（嵌套的）dict，创建一个可变的副本。

参数

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 中添加或替换的键值对字典

返回

一个包含新增和/或替换条目的新字典。

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 是任何其他数据类型，此函数将返回 repr(x)。

参数

• x – 要表示的字典

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

返回

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