nn.Embedding() 是 PyTorch 中用于创建词嵌入层的类。这个层可以将整数类型的词汇表索引映射成密集的实数向量表示。它的用法如下:
- 首先创建一个 Embedding 对象:
embedding_layer = nn.Embedding(vocab_size, embedding_dim)
nn.Embedding() 是 PyTorch 中用于创建词嵌入层的类。这个层可以将整数类型的词汇表索引映射成密集的实数向量表示。它的用法如下:
embedding_layer = nn.Embedding(vocab_size, embedding_dim)