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 没有版本计数器）。