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, optional) – 保留屬性 tensor_name 的引數化。預設為 True

返回

模組

返回型別

模組

引發

• ValueError – 如果 module[tensor_name] 未被引數化

• ValueError – 如果 leave_parametrized=False 且引數化依賴於多個張量