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 中的未参数化张量。这仅当参数化只依赖于一个张量时才可行。

参数:

• 模块（nn.Module）- 从中移除参数化的模块

• tensor_name（字符串）- 要移除的参数化名称

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

返回值:

模块

返回类型:

模块

引发:

• ValueError – 如果 module[tensor_name] 没有参数化

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