torch.nn.modules.sparse.EmbeddingBag Class Reference

Inheritance diagram for torch.nn.modules.sparse.EmbeddingBag:

Public Member Functions
def	__init__ (self, num_embeddings, embedding_dim, max_norm=None, norm_type=2., scale_grad_by_freq=False, mode='mean', sparse=False, _weight=None)
def	reset_parameters (self)
def	forward (self, input, offsets=None)
def	extra_repr (self)
def	from_pretrained (cls, embeddings, freeze=True, max_norm=None, norm_type=2., scale_grad_by_freq=False, mode='mean', sparse=False)

Public Attributes
	num_embeddings
	embedding_dim
	max_norm
	norm_type
	scale_grad_by_freq
	weight
	mode
	sparse

Detailed Description

Definition at line 177 of file sparse.py.

---

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

• torch/nn/modules/sparse.py