评价此页

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

返回类型

模块