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