参数化列表

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

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

当 module[tensor_name] 已经被 register_parametrization() 参数化时，它是 module.parametrizations[tensor_name] 的类型。

如果第一个注册的参数化有一个返回一个张量的right_inverse，或者没有right_inverse（在这种情况下我们假设right_inverse是恒等函数），它将持有名为original的张量。如果它有一个返回多个张量的right_inverse，这些将被注册为original0，original1，…

警告

此类由 register_parametrization() 在内部使用。此处对其进行文档化以确保完整性。用户不应实例化它。

Parameters

• 模块 (序列) – 表示参数化的模块序列

• 原始 (Parameter 或 Tensor) – 参数化参数或缓冲区

• 不安全 (bool) – 一个布尔标志，表示参数化是否可能改变张量的dtype和形状。默认值：False 警告：在注册时不会检查参数化的一致性。启用此标志需自行承担风险。

right_inverse(value)

按照反向注册顺序调用参数化中的 right_inverse 方法。

然后，如果 right_inverse 输出一个张量，则将结果存储在 self.original 中，或者如果它输出多个，则存储在 self.original0、self.original1、… 中。

Parameters

值 (张量) – 初始化模块的值