pytorch实现word embedding: nn.Embedding

pytorch中实现词嵌入的模块是torch.nn.Embedding(m,n),其中m是单词总数,n是单词的特征属性数目。例如:

embedding = nn.Embedding(10, 3)  #总共有10个单词,每个单词表示为3个维度特征。此行程序创建了一个10*3的随机矩阵,每一行代表一个单词

embedding的初始化

embedding的读取

embedding的权重的训练和更新

猜你喜欢

转载自www.cnblogs.com/picassooo/p/12535916.html