torch.autograd.graph.increment_version

torch.autograd.graph.increment_version(tensor)

更新 autograd 元資料，跟蹤給定 Tensor 是否被原地（in place）修改。

這樣做是為了在 autograd 引擎中啟用更精確的錯誤檢查。PyTorch 函式在適當呼叫 mark_dirty() 時已自動完成此操作，因此只有當你以 PyTorch 不知道的方式對 Tensor 資料進行原地（inplace）操作時，才需要顯式呼叫此函式。例如，一個自定義核函式讀取 Tensor 的 data_ptr 並根據此指標原地修改記憶體。可以接受單個 Tensor 或 Tensor 列表。

請注意，對於單個原地操作多次增加版本計數器並無問題。

請注意，如果你傳入在 torch.inference_mode() 下構造的 tensor，我們不會增加其版本計數器（因為你的 tensor 沒有版本計數器）。