評價此頁

torch.autograd.graph.increment_version#

torch.autograd.graph.increment_version(tensor)[原始碼]#

更新 autograd 元資料,跟蹤給定的 Tensor 是否被原地修改。

這是為了實現 autograd 引擎內更準確的錯誤檢查。PyTorch 函式和自定義 Function 在呼叫 `mark_dirty()` 時已經自動完成此操作,因此只有當您以 PyTorch 不知曉的方式對 Tensor 資料進行原地操作時,才需要顯式呼叫此函式。例如,一個讀取 Tensor `data_ptr` 並根據此指標原地修改記憶體的自定義核心。可以接受一個 Tensor 或一個 Tensor 列表。

請注意,對單個原地操作多次遞增版本計數器不會有問題。

請注意,如果您傳入在 `torch.inference_mode()` 下構造的 Tensor,我們將不會增加其版本計數器(因為您的 Tensor 沒有版本計數器)。