其他分享
首页 > 其他分享> > nn.Embedding

nn.Embedding

作者:互联网

torch.nn.Embedding(num_embeddings, embedding_dim, padding_idx=None,
 max_norm=None,  norm_type=2.0,   scale_grad_by_freq=False, 
 sparse=False,  _weight=None)

其为一个简单的存储固定大小的词典的嵌入向量的查找表,意思就是说,给一个编号,嵌入层就能返回这个编号对应的嵌入向量,嵌入向量反映了各个编号代表的符号之间的语义关系。

输入为一个编号列表,输出为对应的符号嵌入向量列表。

实际使用参考:

https://www.jianshu.com/p/63e7acc5e890

标签:嵌入,nn,python,向量,Embedding,范数,optional,norm
来源: https://www.cnblogs.com/lishijun/p/15785881.html