Список модулей

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

Содержит подмодули в списке.

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

Параметры:

modules (iterable, необязательно) – итерируемый объект модулей для добавления

Пример:

class MyModule(nn.Module):
    def __init__(self):
        super(MyModule, self).__init__()
        self.linears = nn.ModuleList([nn.Linear(10, 10) for i in range(10)])

    def forward(self, x):
        # ModuleList can act as an iterable, or be indexed using ints
        for i, l in enumerate(self.linears):
            x = self.linears[i // 2](x) + l(x)
        return x

append(module) [source]

Добавляет данный модуль в конец списка.

Параметры:

module (nn.Module) – добавляемый модуль

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

СписокМодулей

extend(modules) [source]

Добавляет модули из итерируемого Python объекта в конец списка.

Параметры:

modules (iterable) – итерируемый объект модулей для добавления

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

СписокМодулей

insert(index, module) [source]

Вставляет данный модуль перед заданным индексом в списке.

Параметры:

• index (int) – индекс для вставки.
• module (nn.Module) – модуль для вставки