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

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

例如，輸入10個詞，每個詞用2維向量表征，對應的權重就是一個10x2的矩陣。

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

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。