torch.nn.utils.parametrize.remove_parametrizations

torch.nn.utils.parametrize.remove_parametrizations(module, tensor_name, leave_parametrized=True)

移除模組中 tensor 的參數化 (parametrizations)。

• 如果 leave_parametrized=True，則 module[tensor_name] 將會被設定為它目前的輸出。在這種情況下，參數化不應該改變 tensor 的 dtype。

• 如果 leave_parametrized=False，則 module[tensor_name] 將會被設定為 module.parametrizations[tensor_name].original 中的未參數化的 tensor。 只有當參數化只依賴一個 tensor 時，才有可能這樣做。

參數

• module (nn.Module) – 從中移除參數化的模組

• tensor_name (str) – 要移除的參數化的名稱

• leave_parametrized (bool, optional) – 保持 tensor_name 屬性參數化。預設值： True

返回

module

返回類型

Module

引發

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

• ValueError – 如果 leave_parametrized=False 並且參數化依賴於多個 tensor