Список параметризаций
-
class torch.nn.utils.parametrize.ParametrizationList(modules, original, unsafe=False)[source] -
Последовательный контейнер, который хранит и управляет параметрами или буферами
originalилиoriginal0,original1, … параметризованного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 (Параметр или Тензор) – параметр или буфер, который параметризован
-
unsafe (bool) – флаг, указывающий, может ли параметризация изменить тип и форму тензора. По умолчанию:
FalseПредупреждение: параметризация не проверяется на согласованность при регистрации. Включайте этот флаг на свой страх и риск.
-
right_inverse(value)[source] -
Вызывает методы
right_inverse(см.register_parametrization()) параметризаций в обратном порядке их регистрации. Затем результат хранится вself.original, еслиright_inverseвозвращает один тензор, или вself.original0,self.original1, … если возвращает несколько.- Параметры
-
value (Тензор) – Значение, с которым инициализировать модуль
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.nn.utils.parametrize.ParametrizationList.html