参数化列表 ¶
- class torch.nn.utils.parametrize.ParametrizationList(modules, original, unsafe=False)[source][source]¶
一个顺序容器,用于持有和管理参数化的原始参数或缓冲区。
它是当
module[tensor_name]
使用register_parametrization()
进行参数化时的module.parametrizations[tensor_name]
类型。如果第一个注册的参数化具有返回一个张量或没有
right_inverse
(在这种情况下,我们假设right_inverse
是恒等映射),它将使用original
的名称保存张量。如果它有一个返回多个张量的right_inverse
,这些张量将被注册为original0
、original1
等。警告
此类由
register_parametrization()
内部使用。此处记录以供完整性。用户不应实例化此类。- 参数:
模块(序列)- 表示参数化的模块序列
原始(参数或张量)- 被参数化的参数或缓冲区
unsafe (bool) – 表示参数化是否可能更改张量的数据类型和形状的布尔标志。默认:False 警告:注册时不会检查参数化的一致性。启用此标志需自行承担风险。
- right_inverse(value)[source][source]
按逆注册顺序调用参数化的
right_inverse
方法。然后,如果
right_inverse
输出一个张量,则将其存储在self.original
中;如果它输出多个,则存储在self.original0
、self.original1
、… 中。- 参数:
模块初始化值(张量)- 初始化模块的值