torch.autograd.grad_mode.set_grad_enabled Class Reference

Inheritance diagram for torch.autograd.grad_mode.set_grad_enabled:

Public Member Functions
def	__init__ (self, mode)
def	__enter__ (self)
def	__exit__ (self, args)

Public Attributes
	prev

Detailed Description

Definition at line 91 of file grad_mode.py.

---

The documentation for this class was generated from the following file:

• torch/autograd/grad_mode.py