Список параметров

class torch.nn.ParameterList(values=None) [source]

Содержит параметры в списке.

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

Обратите внимание, что конструктор, присвоение элемента списка, append() метод и extend() метод преобразуют любой Tensor в Parameter.

Параметры:

parameters (iterable, optional) – последовательность элементов для добавления в список.

Пример:

class MyModule(nn.Module):
    def __init__(self):
        super(MyModule, self).__init__()
        self.params = nn.ParameterList([nn.Parameter(torch.randn(10, 10)) for i in range(10)])

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

append(value) [source]

Добавляет заданное значение в конец списка.

Параметры:

value (Any) – значение для добавления

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

Список параметров

extend(values) [source]

Добавляет значения из Python-последовательности в конец списка.

Параметры:

values (iterable) – последовательность значений для добавления

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

Список параметров