Search notes:

torch.nn.Embedding

torch.nn.Embedding is a lookup table that stores embeddings of a fixed dictionary and size.
torch.nn.Embedding inherits from torch.nn.Module.

Links

The source code is in torch/nn/modules/sparse.py.

Index