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