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
- 返回类型
- 引发
ValueError – 如果
module[tensor_name]
未参数化ValueError – 如果
leave_parametrized=False
且参数化依赖于多个张量