Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.autograd.function.NestedIOFunction Class Reference
Inheritance diagram for torch.autograd.function.NestedIOFunction:
torch.autograd.function.Function torch.autograd.function.FunctionMeta torch.autograd.function._ContextMethodMixin torch.autograd.function._HookMixin

Public Member Functions

def backward (self, gradients)
def forward (self, args)
def save_for_backward (self, args)
def saved_tensors (self)
def mark_dirty (self, args, kwargs)
def mark_non_differentiable (self, args, kwargs)
def forward_extended (self, input)
def backward_extended (self, grad_output)
- Public Member Functions inherited from torch.autograd.function.FunctionMeta
def __init__ (cls, name, bases, attrs)
- Public Member Functions inherited from torch.autograd.function._ContextMethodMixin
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

- Public Attributes inherited from torch.autograd.function._ContextMethodMixin

Additional Inherited Members

- Static Public Member Functions inherited from torch.autograd.function.Function
def forward (ctx, args, kwargs)
def backward (ctx, grad_outputs)
- Static Public Attributes inherited from torch.autograd.function.Function

Detailed Description

Definition at line 332 of file

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