embedding = nn.Embedding(num_embeddings, embedding_dim)
  • num_embeddings: 词表大小
  • embedding_dim: 词向量长度
  • 本质上讲,是存了一个二维矩阵,对于输入的词的索引返回对应的行