torch.autograd.graph.increment_version¶ torch.autograd.graph.increment_version(tensor)[源代码]¶ 更新 autograd 元数据,跟踪给定的 Tensor 是否在原地进行了修改。 这是为了在自动求导引擎中实现更准确的错误检查。 PyTorch函数和自定义函数中已经自动完成了此操作,当适当调用mark_dirty()时,因此您只需要在Pytorch不知道的情况下对Tensor数据进行就地操作时显式调用此函数。例如,一个自定义内核读取Tensor data_ptr并基于此指针就地修改内存。 请注意,对于单个就地操作多次递增版本计数器不会有问题。