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().__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]