Embedding — PyTorch 1.10.1 documentation
pytorch.org › generated › torchA simple lookup table that stores embeddings of a fixed dictionary and size. This module is often used to store word embeddings and retrieve them using indices. The input to the module is a list of indices, and the output is the corresponding word embeddings. Parameters. num_embeddings ( int) – size of the dictionary of embeddings.