評價此頁

torch.nn.utils.parametrize.transfer_parametrizations_and_params#

torch.nn.utils.parametrize.transfer_parametrizations_and_params(from_module, to_module, tensor_name=None)[原始碼]#

將引數化和它們所引數化的引數從 from_module 轉移到 to_module

如果指定了 tensor_name,則只轉移指定的引數;否則,轉移所有引數化的引數。如果 to_module 中不存在這些引數,則會建立它們。如果 from_module 未被引數化,則不做任何操作。

引數
  • from_module (nn.Module) – 要從中轉移的模組

  • to_module (nn.Module) – 要轉移到的模組

  • tensor_name (str, optional) – 要轉移的引數

返回

to_module

返回型別

模組