pytorch中，針對詞向量有一個專門的層nn.Embedding，用來實現(xiàn)詞與詞向量的映射。

nn.Embedding具有一個權(quán)重，形狀是(num_embeddings,embedding_dimension)。

例如，輸入10個詞，每個詞用2維向量表征，對應(yīng)的權(quán)重就是一個10x2的矩陣。

如果Embedding層的輸入形狀為NxM（比如N是batch size,M是序列的長度），則輸出的形狀是N*M*embedding_dimension.

《新程序員》：云原生和全面數(shù)字化實踐50位技術(shù)專家共同創(chuàng)作，文字、視頻、音頻交互閱讀

總結(jié)

以上是生活随笔為你收集整理的pytorch 笔记： torch.nn.Embedding的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。