СписокПараметризаций
-
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