Spec-Zone.ru › PyTorch 1

СписокПараметризаций

class torch.nn.utils.parametrize.ParametrizationList(modules, original, unsafe=False) [source]

Последовательный контейнер, который хранит и управляет параметрами или буферами параметризованного torch.nn.Module, такими как original или original0, original1, …

Это тип module.parametrizations[tensor_name], когда module[tensor_name] был параметризован с помощью register_parametrization().

Если первая зарегистрированная параметризация имеет right_inverse, которая возвращает один тензор, или не имеет right_inverse (в этом случае предполагается, что right_inverse — это тождественное преобразование), она будет хранить тензор под именем original. Если у неё есть right_inverse, которая возвращает более одного тензора, эти тензоры будут зарегистрированы как original0, original1, …

Предупреждение

Этот класс используется внутри register_parametrization(). Он документирован здесь для полноты. Его не следует создавать пользователем.

Параметры:
  • modules (sequence) – последовательность модулей, представляющих параметризации
  • original (Parameter или Tensor) – параметр или буфер, который параметризован
  • unsafe (bool) – флаг булевого типа, указывающий, может ли параметризация изменить тип и форму тензора. По умолчанию: False Предупреждение: параметризация не проверяется на согласованность при регистрации. Включите этот флаг на свой страх и риск.
right_inverse(value) [source]

Вызывает методы right_inverse (см. register_parametrization()) параметризаций в обратном порядке их регистрации. Затем результат сохраняется в self.original если right_inverse возвращает один тензор или в self.original0, self.original1, … если возвращает несколько.

Параметры:

value (Tensor) – Значение, с которым инициализировать модуль

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.nn.utils.parametrize.ParametrizationList.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API