ParameterDict

class torch.nn.ParameterDict(parameters=None)

在一個字典中儲存引數。

ParameterDict 可以像普通的 Python 字典一樣進行索引，但其包含的 Parameters 會被正確註冊，並對所有 Module 方法可見。其他物件則像普通 Python 字典一樣處理。

ParameterDict 是一個有序字典。update() 方法與其他無序對映型別（例如 Python 的普通 dict）一起使用時，不會保留合併對映的順序。另一方面，OrderedDict 或另一個 ParameterDict 將保留其順序。

請注意，建構函式、字典元素的賦值以及 update() 方法會將任何 Tensor 轉換為 Parameter。

引數

values (iterable, optional) – (string : Any) 的對映（字典）或型別為 (string, Any) 的鍵值對可迭代物件

示例

class MyModule(nn.Module):
    def __init__(self) -> None:
        super().__init__()
        self.params = nn.ParameterDict({
                'left': nn.Parameter(torch.randn(5, 10)),
                'right': nn.Parameter(torch.randn(5, 10))
        })

    def forward(self, x, choice):
        x = self.params[choice].mm(x)
        return x

clear()

從 ParameterDict 中移除所有項。

copy()

返回此 ParameterDict 例項的副本。

返回型別

ParameterDict

fromkeys(keys, default=None)

返回一個包含提供的鍵的新 ParameterDict。

引數

• keys (iterable, string) – 用於建立新 ParameterDict 的鍵

• default (Parameter, optional) – 為所有鍵設定的值

返回型別

ParameterDict

get(key, default=None)

如果存在，則返回與 key 關聯的引數。否則，如果提供了 default，則返回 default；如果未提供，則返回 None。

引數

• key (str) – 從 ParameterDict 中獲取的鍵

• default (Parameter, optional) – 如果 key 不存在，則返回的值

返回型別

Any

items()

返回 ParameterDict 鍵值對的可迭代物件。

返回型別

Iterable[tuple[str, Any]]

keys()

返回 ParameterDict 鍵的可迭代物件。

返回型別

Iterable[str]

pop(key)

從 ParameterDict 中移除 key 並返回其引數。

引數

key (str) – 從 ParameterDict 中彈出的鍵

返回型別

Any

popitem()

移除並返回 ParameterDict 中最後插入的 (key, parameter) 對。

返回型別

tuple[str, Any]

setdefault(key, default=None)

為 ParameterDict 中的 key 設定預設值。

如果 key 在 ParameterDict 中，則返回其值。否則，使用引數 default 插入 key 並返回 default。default 預設為 None。

引數

• key (str) – 要設定預設值的鍵

• default (Any) – 設定給鍵的引數

返回型別

Any

update(parameters)

使用來自 parameters 的鍵值對更新 ParameterDict，覆蓋現有鍵。

注意

如果 parameters 是一個 OrderedDict、一個 ParameterDict 或一個鍵值對可迭代物件，則其中新元素的順序將被保留。

引數

parameters (iterable) – 從字串到 Parameter 的對映（字典），或型別為 (string, Parameter) 的鍵值對可迭代物件

values()

返回 ParameterDict 值的可迭代物件。

返回型別

Iterable[Any]]