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, необязательный) – параметр для переноса

Возвращает:

to_module

Тип возвращаемого значения:

Module