参数列表

classtorch.nn.utils.parametrize.ParametrizationList(modules, original, unsafe=False)[源代码]

一个顺序容器，用于存放和管理参数化 torch.nn.Module 的原始参数或缓冲区。

当使用register_parametrization() 对 module[tensor_name] 进行参数化时，module.parametrizations[tensor_name] 的类型。

如果第一个注册的参数化具有一个返回单个张量的right_inverse，或者它没有right_inverse（在这种情况下我们假设right_inverse是恒等变换），则将该张量以名称original保存。如果它的right_inverse返回多个张量，则这些张量将以original0, original1, …的形式注册。

警告

此类由 register_parametrization() 内部使用，并在此进行文档说明以确保完整性。用户不应实例化此类。

参数

• modules (序列) – 代表参数化的一系列模块

• original (Parameter 或 Tensor) – 被参数化的参数或张量

• unsafe (bool) – 一个布尔标志，表示参数化是否可能更改张量的数据类型和形状。默认值为 False。警告：在注册时不会验证参数化的连贯性，请自行承担启用此标志的风险。

right_inverse(value)[源代码]

按照逆注册顺序调用各参数化对象的right_inverse方法。

然后，将结果存储在 self.original 中，如果 right_inverse 输出一个张量；如果输出多个张量，则分别存储在 self.original0、self.original1 等中。

参数

value (Tensor) – 模块的初始化值