torch.Tensor.module_load

Tensor.module_load(other, assign=False) [source]

Определяет, как преобразовать other при загрузке в self в load_state_dict().

Используется, когда get_swap_module_params_on_conversion() равно True.

Предполагается, что self является параметром или буфером в nn.Module, а other — значением в словаре состояния с соответствующим ключом. Этот метод определяет, как other переназначается перед обменом с self посредством swap_tensors() в load_state_dict().

Примечание

Этот метод всегда должен возвращать новый объект, отличный от self и other. Например, реализация по умолчанию возвращает self.copy_(other).detach(), если assign — это False, или other.detach(), если assign — это True.

Параметры:

• other (Tensor) – значение в словаре состояния с ключом, соответствующим self
• assign (bool) – аргумент assign, переданный в nn.Module.load_state_dict()