Spec-Zone.ru › PyTorch 2

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

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

Spec-Zone.ru

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