МодульModuleDict

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

Хранит подмодули в словаре.

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

ModuleDict является упорядоченным словарем, который учитывает

• порядок вставки, и
• в update(), порядок объединённых OrderedDict, dict (начав с Python 3.6) или другого ModuleDict (аргумент для update()).

Обратите внимание, что update() со словарями, не имеющими порядка (например, простые словари Python до версии 3.6) не сохраняет порядок объединённого словаря.

Параметры:

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

Пример:

class MyModule(nn.Module):
    def __init__(self):
        super(MyModule, self).__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() [source]

Удаляет все элементы из ModuleDict.

items() [source]

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

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

Iterable[Tuple[str, Модуль]]

keys() [source]

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

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

Iterable[str]

pop(key) [source]

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

Параметры:

key (str) – ключ для удаления из ModuleDict

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

Модуль

update(modules) [source]

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

Примечание

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

Параметры:

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

values() [source]

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

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

Iterable[Модуль]