Embedding — PyTorch 1.10.1 documentation
pytorch.org › generated › torchThis 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 embedding_dim ( int) – the size of each embedding vector