ParameterDict

class torch.nn.ParameterDict(parameters=None)

在字典中保存參數。

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

ParameterDict 是一個有序字典。使用其他無序映射類型（例如，Python 的普通 dict）進行 update() 不會保留合併映射的順序。另一方面，OrderedDict 或另一個 ParameterDict 將保留它們的順序。

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

參數

values (iterable, optional) – (字串 : Any) 的映射（字典），或是 (字串, 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)

如果存在，則傳回與鍵相關聯的參數。否則，如果提供了 default，則傳回 default，否則傳回 None。

參數

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

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

傳回類型

Any

items()

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

傳回類型

Iterable[Tuple[str, Any]]

keys()

傳回 ParameterDict 鍵的可迭代物件。

傳回類型

Iterable[str]

pop(key)

從 ParameterDict 中移除鍵並傳回其參數。

參數

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

傳回類型

Any

popitem()

從 ParameterDict 中移除並傳回最後插入的 (鍵, 參數) 對。

傳回類型

Tuple[str, Any]

setdefault(key, default=None)

設定 ParameterDict 中某個鍵的預設值。

如果鍵在 ParameterDict 中，則傳回其值。如果沒有，則插入具有參數 default 的 key 並傳回 default。default 預設為 None。

參數

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

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

傳回類型

Any

update(parameters)

使用 parameters 中的鍵值對更新 ParameterDict，覆寫現有的鍵。

注意

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

參數

parameters (iterable) – 從字串到 Parameter 的映射（字典），或是 (字串, Parameter) 類型的鍵值對的可迭代對象

values()

傳回 ParameterDict 值的可迭代對象。

傳回類型

Iterable[Any]