Python PyTorch ParameterDict用法及代碼示例

本文簡要介紹python語言中 torch.nn.ParameterDict 的用法。

用法:

class torch.nn.ParameterDict(parameters=None)

參數：

parameters(可迭代的,可選的) -(字符串：Parameter)的映射(字典)或類型為(字符串，Parameter)的鍵值對的可迭代

在字典中保存參數。

ParameterDict 可以像普通的 Python 字典一樣被索引，但它包含的參數已正確注冊，並且對所有 Module 方法都是可見的。

ParameterDict是一個排序尊重的字典

• 插入順序，以及

• 在 update() 中，合並的 OrderedDict 或另一個 ParameterDict 的順序(update() 的參數)。

請注意，update() 與其他無序映射類型(例如，Python 的普通 dict )不會保留合並映射的順序。

例子：

class MyModule(nn.Module):
    def __init__(self):
        super(MyModule, self).__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