ModuleDict

class torch.nn.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]