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:

• torch/nn/modules/sparse.py