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

返回值

模組

返回值型別

Module

丟擲

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

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