torch.autograd.function._ContextMethodMixin Class Reference

Inheritance diagram for torch.autograd.function._ContextMethodMixin:

Public Member Functions
def	save_for_backward (self, tensors)
def	mark_dirty (self, args)
def	mark_shared_storage (self, pairs)
def	mark_non_differentiable (self, args)

Public Attributes
	to_save
	dirty_tensors
	non_differentiable

Detailed Description

Definition at line 10 of file function.py.

---

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

• torch/autograd/function.py