torch.Tensor.module_load

Tensor.module_load(other, assign=False)

定义在 load_state_dict() 中将 other 加载到 self 时如何对其进行转换。

当 get_swap_module_params_on_conversion() 为 True 时使用。

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

注意

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

参数

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

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