torch.autograd.graph.increment_version

torch.autograd.graph.increment_version(tensor)

更新 autograd 元數據，追蹤給定的 Tensor 是否已就地修改。

這是為了在 autograd 引擎中啟用更準確的錯誤檢查。 它已經由 PyTorch 函數自動完成，並在適當地呼叫 mark_dirty() 時在自訂 Function 中完成，因此只有當您以 PyTorch 不知道的方式對 Tensor 資料執行就地操作時，才需要明確地呼叫此函數。 例如，一個自訂核心讀取 Tensor data_ptr 並基於此指標就地修改記憶體。 可以接受一個張量或一個張量列表。

請注意，對於單個就地操作，多次遞增版本計數器沒有問題。

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