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 參數