ModuleDict

class torch.nn.ModuleDict(modules=None)

在字典中保存子模块。

ModuleDict 可以像常规 Python 字典一样进行索引，但它包含的模块已正确注册，并且对于所有 Module 方法都是可见的。

ModuleDict 是一个有序字典，它遵循

• 插入顺序，以及

• 在 update() 中，合并的 OrderedDict、dict（从 Python 3.6 开始）或另一个 ModuleDict（update() 的参数）的顺序。

请注意，使用其他无序映射类型（例如，Python 3.6 版本之前的普通 dict）进行 update() 不会保留合并映射的顺序。

参数

modules (iterable, optional) – (字符串: 模块) 的映射（字典）或 (字符串, 模块) 类型的键值对的可迭代对象

示例

class MyModule(nn.Module):
    def __init__(self) -> None:
        super().__init__()
        self.choices = nn.ModuleDict({
                'conv': nn.Conv2d(10, 10, 3),
                'pool': nn.MaxPool2d(3)
        })
        self.activations = nn.ModuleDict([
                ['lrelu', nn.LeakyReLU()],
                ['prelu', nn.PReLU()]
        ])

    def forward(self, x, choice, act):
        x = self.choices[choice](x)
        x = self.activations[act](x)
        return x

clear()

从 ModuleDict 中删除所有项。

items()

返回 ModuleDict 键/值对的可迭代对象。

返回类型

Iterable[Tuple[str, Module]]

keys()

返回 ModuleDict 键的可迭代对象。

返回类型

Iterable[str]

pop(key)

从 ModuleDict 中删除键并返回其模块。

参数

key (str) – 要从 ModuleDict 中弹出的键

返回类型

Module

update(modules)

使用来自映射的键值对更新 ModuleDict，覆盖现有键。

注意

如果 modules 是 OrderedDict、ModuleDict 或键值对的可迭代对象，则会保留其中新元素的顺序。

参数

modules (iterable) – 从字符串到 Module 的映射（字典），或 (字符串, Module) 类型的键值对的可迭代对象

values()

返回 ModuleDict 值的可迭代对象。

返回类型

Iterable[Module]