Сохранить и восстановить состояние автограда

Question

Есть ли способ сохранить и восстановить текущее состояние автограда?

Например:

saved_state = torch.autograd.get_grad_enabled()  # How do I do this?
torch.autograd.set_grad_enabled(temporary_state)
# do things for great justice
torch.autograd.set_grad_enabled(saved_state)

Answer 1

Исходный код для torch.autograd.grad_mode намекает на решение:

torch.is_grad_enabled()

Похоже, эта функция в настоящее время недокументирована .