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

Public Member Functions

def forward (cls, ctx, weight, indices, offsets, max_norm, norm_type, scale_grad_by_freq, mode)
 
- 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)
 

Static Public Member Functions

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

Additional Inherited Members

- Public Attributes inherited from torch.autograd.function._ContextMethodMixin
 to_save
 
 dirty_tensors
 
 non_differentiable
 
- Static Public Attributes inherited from torch.autograd.function.Function
 is_traceable
 

Detailed Description

Definition at line 13 of file sparse.py.


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