ParameterDict

class torch.nn.modules.container.ParameterDict(parameters=None) [исходный код]

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

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

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

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

Параметры:

values (итерируемый объект, необязательно) – отображение (словарь) пар (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 (итерируемый объект, строка) – ключи для создания нового ParameterDict
• default (Parameter, необязательно) – значение, устанавливаемое для всех ключей

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

ParameterDict

get(key, default=None) [исходный код]

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

Параметры:

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

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

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 (итерируемый объект) – отображение (словарь) строк в Parameter или итерируемый объект пар «ключ-значение» типа (string, Parameter)

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

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

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

Iterable[Any]