ModuleDict

class torch.nn.modules.container.ModuleDict(modules=None) [исходный код]

Содержит вложенные модули в словаре.

ModuleDict можно индексировать как обычный словарь Python, но содержащиеся в нём модули регистрируются должным образом и доступны всем методам Module.

ModuleDict — это упорядоченный словарь, который сохраняет

• порядок добавления элементов и
• при вызове update() — порядок объединённого OrderedDict, dict или другого ModuleDict (аргумента для update()).

Обратите внимание, что вызов update() с другими неупорядоченными типами отображений не сохраняет порядок объединённого отображения.

Параметры:

modules (итерируемый объект, необязательно) – отображение (словарь) вида (строка: модуль) или итерируемый объект с парами ключ-значение типа (строка, модуль)

Пример:

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.

Тип возвращаемого значения:

ItemsView[str, Module]

keys() [исходный код]

Возвращает итерируемый объект с ключами ModuleDict.

Тип возвращаемого значения:

KeysView[str]

pop(key) [исходный код]

Удаляет ключ из ModuleDict и возвращает соответствующий модуль.

Параметры:

key (str) – ключ, который нужно удалить из ModuleDict

Тип возвращаемого значения:

Module

update(modules) [исходный код]

Обновляет ModuleDict парами ключ-значение из отображения, перезаписывая существующие ключи.

Примечание

Если modules является OrderedDict, объектом ModuleDict или итерируемым объектом с парами ключ-значение, порядок новых элементов в нём сохраняется.

Параметры:

modules (итерируемый объект) – отображение (словарь) строк в объекты Module или итерируемый объект с парами ключ-значение типа (строка, Module)

values() [исходный код]

Возвращает итерируемый объект со значениями ModuleDict.

Тип возвращаемого значения:

ValuesView[Module]