ParameterList

class torch.nn.ParameterList(values=None) [исходный код]

Хранит параметры в списке.

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

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

Параметры:

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

Пример:

class MyModule(nn.Module):
    def __init__(self) -> None:
        super().__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) [исходный код]

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

Параметры:

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

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

Self

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

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

Параметры:

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

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

Self

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

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

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

str