torch.autograd.graph.increment_version#
- torch.autograd.graph.increment_version(tensor)[源代码]#
更新 autograd 元数据,跟踪给定的 Tensor 是否被原地修改。
这是为了实现 autograd 引擎内更准确的错误检查。PyTorch 函数和自定义 Function 在调用 `mark_dirty()` 时已经自动完成此操作,因此只有当您以 PyTorch 不知晓的方式对 Tensor 数据进行原地操作时,才需要显式调用此函数。例如,一个读取 Tensor `data_ptr` 并根据此指针原地修改内存的自定义内核。可以接受一个 Tensor 或一个 Tensor 列表。
请注意,对单个原地操作多次递增版本计数器不会有问题。
请注意,如果您传入在 `torch.inference_mode()` 下构造的 Tensor,我们将不会增加其版本计数器(因为您的 Tensor 没有版本计数器)。