快捷键

参数化列表 ¶

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 ,这些张量将被注册为 original0original1 等。

警告

此类由 register_parametrization() 内部使用。此处记录以供完整性。用户不应实例化此类。

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

  • 原始(参数或张量)- 被参数化的参数或缓冲区

  • unsafe (bool) – 表示参数化是否可能更改张量的数据类型和形状的布尔标志。默认:False 警告:注册时不会检查参数化的一致性。启用此标志需自行承担风险。

right_inverse(value)[source][source]

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

然后,如果 right_inverse 输出一个张量,则将其存储在 self.original 中;如果它输出多个,则存储在 self.original0self.original1 、… 中。

参数:

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


© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,并使用 Read the Docs 提供的主题。

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源