保存并恢复autograd启用状态

Question

有没有办法保存和恢复autograd的当前状态？

例如：

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

Source code for torch.autograd.grad_mode暗示了一种解决方案：

torch.is_grad_enabled()

看来this function is currently undocumented. .。