ParametrizationList

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

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

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

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

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

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

Параметры:

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

right_inverse(value) [source]

Вызывает методы right_inverse параметризаций в обратном порядке их регистрации.

Затем результат сохраняется в self.original, если right_inverse возвращает один тензор, или в self.original0, self.original1, …, если возвращается несколько тензоров.

Параметры:

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