ParameterDict

class torch.nn.ParameterDict(parameters=None) [source]

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

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

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

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

Параметры:

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

Пример:

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

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

copy() [source]

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

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

ParameterDict

fromkeys(keys, default=None) [source]

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

Параметры:

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

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

ParameterDict

get(key, default=None) [source]

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

Параметры:

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

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

Any

items() [source]

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

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

Iterable[Tuple[str, Any]]

keys() [source]

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

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

Iterable[str]

pop(key) [source]

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

Параметры:

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

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

Any

popitem() [source]

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

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

Tuple[str, Any]

setdefault(key, default=None) [source]

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

Параметры:

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

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

Any

update(parameters) [source]

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

Примечание

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

Параметры:

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

values() [source]

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

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

Iterable[Any]