ModuleDict#

class torch.nn.ModuleDict(modules=None)[source]#

以字典形式儲存子模組。

ModuleDict 可以像普通 Python 字典一樣索引，但其中包含的模組會被正確註冊，並且對所有 Module 方法可見。

ModuleDict 是一個**有序**字典，它遵循

插入順序，並且
在 update() 中，會保持合併的 OrderedDict、dict (從 Python 3.6 開始) 或另一個 ModuleDict (作為 update() 的引數) 的順序。

請注意，使用其他無序對映型別 (例如，Python 3.6 之前的普通 dict) 呼叫 update() 不會保留合併對映的順序。

引數: modules (iterable, optional) – 一個 (string: module) 的對映 (字典) 或一個 (string, module) 型別鍵值對的可迭代物件

示例

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