torch.nn.functional.embedding — PyTorch 1.10.1 documentation
pytorch.org › torchtorch.nn.functional.embedding(input, weight, padding_idx=None, max_norm=None, norm_type=2.0, scale_grad_by_freq=False, sparse=False) [source] A simple lookup table that looks up embeddings in a fixed dictionary and size. This module is often used to retrieve word embeddings using indices. The input to the module is a list of indices, and the ...
Embedding — PyTorch 1.10.1 documentation
pytorch.org › generated › torchEmbedding¶ class torch.nn. Embedding (num_embeddings, embedding_dim, padding_idx = None, max_norm = None, norm_type = 2.0, scale_grad_by_freq = False, sparse = False, _weight = None, device = None, dtype = None) [source] ¶ A simple lookup table that stores embeddings of a fixed dictionary and size.