Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.nn.modules.sparse.Embedding Class Reference
Inheritance diagram for torch.nn.modules.sparse.Embedding:

Public Member Functions

def __init__ (self, num_embeddings, embedding_dim, padding_idx=None, max_norm=None, norm_type=2., scale_grad_by_freq=False, sparse=False, _weight=None)
 
def reset_parameters (self)
 
def forward (self, input)
 
def extra_repr (self)
 
def from_pretrained (cls, embeddings, freeze=True, padding_idx=None, max_norm=None, norm_type=2., scale_grad_by_freq=False, sparse=False)
 

Public Attributes

 num_embeddings
 
 embedding_dim
 
 padding_idx
 
 max_norm
 
 norm_type
 
 scale_grad_by_freq
 
 weight
 
 sparse
 

Detailed Description

Definition at line 11 of file sparse.py.


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