torch.Tensor.module_load

Tensor.module_load(other, assign=False)

定义了在加载到self时如何转换other，在load_state_dict()中。

当 get_swap_module_params_on_conversion() 为 True 时使用。

预计self是nn.Module中的一个参数或缓冲区，而other是状态字典中具有相应键的值，此方法定义了在通过swap_tensors`()与self交换之前如何重新映射other，在module.load_state_dict()中。

注意

此方法应始终返回一个新对象，该对象不是 self 或 other。 例如，默认实现返回 self.copy_(other).detach() 如果 assign 是 False 或 other.detach() 如果 assign 是 True。

Parameters

• 其他 (张量) – 状态字典中与键对应于 self 的值

• 分配 (布尔值) – 传递给 nn.Module.load_state_dict() 的分配参数