torch.autograd.graph.increment_version

torch.autograd.graph.increment_version(tensor) [исходный код]

Обновляет метаданные autograd, отслеживающие, изменялся ли указанный Tensor на месте.

Это позволяет точнее проверять ошибки в движке autograd. PyTorch уже делает это автоматически для своих функций, а также внутри пользовательской Function, если вызов mark_dirty() выполнен должным образом. Поэтому вызывать эту функцию явно нужно только в том случае, если вы выполняете операцию на месте с данными Tensor способом, о котором PyTorch не знает. Например, пользовательское ядро может считывать data_ptr Tensor и изменять память на месте, используя этот указатель. Функция принимает тензор или список тензоров.

Обратите внимание, что многократное увеличение счётчика версий в рамках одной операции на месте не вызывает проблем.

Обратите внимание, что если передать тензор, созданный в режиме torch.inference_mode(), его счётчик версий не будет увеличен (поскольку у такого тензора нет счётчика версий).