ParameterDict

class torch.nn.ParameterDict(parameters=None) [источник]

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

К элементам ParameterDict можно обращаться по индексу, как к элементам обычного словаря Python, но содержащиеся в нём параметры регистрируются должным образом и доступны всем методам Module. Другие объекты обрабатываются так же, как в обычном словаре Python.

ParameterDict — это упорядоченный словарь. update() с другими неупорядоченными типами отображений (например, обычным dict Python) не сохраняет порядок объединённого отображения. С другой стороны, OrderedDict или другой объект ParameterDict сохраняет порядок элементов.

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

Параметры:

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

Пример:

class MyModule(nn.Module):
    def __init__(self) -> None:
        super().__init__()
        self.params = nn.ParameterDict(
            {
                "left": nn.Parameter(torch.randn(5, 10)),
                "right": nn.Parameter(torch.randn(5, 10)),
            }
        )

    def forward(self, x, choice):
        x = self.params[choice].mm(x)
        return x

clear() [источник]

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

copy() [источник]

Возвращает копию этого экземпляра ParameterDict.

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

ParameterDict

fromkeys(keys, default=None) [источник]

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

Параметры:

• keys (iterable, string) – ключи для создания нового ParameterDict
• default (Parameter, optional) – значение, устанавливаемое для всех ключей

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

ParameterDict

get(key, default=None) [источник]

Возвращает параметр, связанный с ключом, если он присутствует. В противном случае возвращает значение по умолчанию, если оно задано, или None, если оно не задано.

Параметры:

• key (str) – ключ для получения из ParameterDict
• default (Parameter, optional) – значение, возвращаемое, если ключ отсутствует

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

Any

items() [источник]

Возвращает итерируемый объект с парами ключ-значение из ParameterDict.

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

Iterable[tuple[str, Any]]

keys() [источник]

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

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

KeysView[str]

pop(key) [источник]

Удаляет ключ из ParameterDict и возвращает связанный с ним параметр.

Параметры:

key (str) – ключ для извлечения из ParameterDict

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

Any

popitem() [источник]

Удаляет и возвращает последнюю добавленную пару (key, parameter) из ParameterDict.

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

tuple[str, Any]

setdefault(key, default=None) [источник]

Задаёт значение по умолчанию для ключа в Parameterdict.

Если ключ присутствует в ParameterDict, возвращает его значение. Если ключа нет, добавляет key с параметром default и возвращает default. default по умолчанию равно None.

Параметры:

• key (str) – ключ, для которого задаётся значение по умолчанию
• default (Any) – параметр, присваиваемый ключу

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

Any

update(parameters) [источник]

Обновляет ParameterDict парами ключ-значение из parameters, заменяя существующие ключи.

Примечание

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

Параметры:

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

values() [источник]

Возвращает итерируемый объект со значениями ParameterDict.

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

Iterable[Any]