flax.core.frozen_dict 套件

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

Python dict 的不可變異體。

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 – 要從 dict 中移除的鍵

回傳

一個包含新的 FrozenDict 和已移除值的配對。

pretty_repr(num_spaces=4)

傳回巢狀字典的縮排表示法。

unfreeze()

解除此 FrozenDict 的凍結狀態。

回傳

此 FrozenDict 實例的解除凍結版本。

flax.core.frozen_dict.freeze(xs)

凍結巢狀 dict。

將巢狀 dict 轉換為 FrozenDict，使其不可變。

參數

xs – 要凍結的字典（一般的 Python dict）。

回傳

已凍結的字典。

flax.core.frozen_dict.unfreeze(x)

解除 FrozenDict 的凍結狀態。

將 FrozenDict 的可變副本轉換為（巢狀）dict，使其可變。

參數

x – 要解除凍結的 Frozen 字典。

回傳

已解除凍結的字典（一般的 Python dict）。

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

建立一個包含新增和/或取代條目的新 dict。這是一個實用函數，可以對 FrozenDict 或一般 dict 進行操作，並模仿 FrozenDict.copy 的行為。

範例

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

參數

• x – 要複製和更新的字典

• add_or_replace – 要在 dict x 中新增或取代的鍵值對的字典

回傳

一個包含新增和/或取代條目的新 dict。

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

建立一個移除一個條目的新 dict。這是一個實用函數，可以對 FrozenDict 或一般 dict 進行操作，並模仿 FrozenDict.pop 的行為。

範例

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

參數

• x – 要從中移除條目的字典

• key – 要從 dict 中移除的鍵

回傳

一個包含新的 dict 和已移除值的配對。

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

傳回巢狀字典的縮排表示法。這是一個實用函數，可以對 FrozenDict 或一般 dict 進行操作，並模仿 FrozenDict.pretty_repr 的行為。如果 x 是任何其他資料類型，此函數將傳回 repr(x)。

參數

• x – 要表示的字典

• num_spaces – 每個縮排層級中的空格字元數

回傳

巢狀字典的縮排字串表示法。