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() 与 self 交换之前如何重映射 other 。

注意

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

参数:

• 其他（Tensor）- 状态字典中与 self 键对应的值

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