torch.Tensor.module_load

Tensor.module_load(other, assign=False)

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

当 get_swap_module_params_on_conversion() 为 True 时使用。

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

注意

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

参数

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

• assign (bool) – 传递给 nn.Module.load_state_dict() 的 assign 参数