• 文档 >
  • torch.nn >
  • torch.nn.utils.parametrize.remove_parametrizations
快捷方式

torch.nn.utils.parametrize.remove_parametrizations

torch.nn.utils.parametrize.remove_parametrizations(module, tensor_name, leave_parametrized=True)[源代码]

移除模块中张量的参数化。

  • 如果 leave_parametrized=True,则 module[tensor_name] 将设置为其当前输出。在这种情况下,参数化不应更改张量的 dtype

  • 如果 leave_parametrized=False,则 module[tensor_name] 将设置为 module.parametrizations[tensor_name].original 中未参数化的张量。只有当参数化仅依赖于一个张量时,这才是可能的。

参数
  • module (nn.Module) – 要从中移除参数化的模块

  • tensor_name (str) – 要移除的参数化的名称

  • leave_parametrized (bool, 可选) – 保留属性 tensor_name 的参数化。默认值:True

返回值

module

返回类型

Module

引发
  • ValueError – 如果 module[tensor_name] 未参数化

  • ValueError – 如果 leave_parametrized=False 且参数化依赖于多个张量

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取适合初学者和高级开发人员的深入教程

查看教程

资源

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

查看资源