torch.autograd.graph.increment_version

torch.autograd.graph.increment_version(tensor)

更新 autograd 元数据跟踪，以指示给定的张量是否已就地修改。

这是为了在 autograd 引擎中启用更准确的错误检查。PyTorch 函数和自定义 Function 已经自动完成此操作，并且在适当调用 mark_dirty() 时也会自动完成，因此只有当您以 PyTorch 不知道的方式对张量数据进行就地操作时，才需要显式调用此函数。例如，一个自定义内核读取张量 data_ptr 并基于此指针就地修改内存。可以接受张量或张量列表。

请注意，对于单个就地操作，多次递增版本计数器没有问题。

请注意，如果您传入在 torch.inference_mode() 下构建的张量，我们将不会增加其版本计数器（因为您的张量没有版本计数器）。