`nn.embedding` is a simple lookup table that stores embeddings of a fixed dictionary and size from pytorch.
An embedding is like a code book. Given the feature is an index in an array of all possible features.
spam fails_task lang_mismatch pii not_appropriate hate_speech sexual_content quality toxicity humor helpfulness creativity violence