embedding的使用

调用 torch.nn.Embedding(m, n) 

m-语料库中字符的个数,例如【‘人’:1,‘们’:2】,m=2

n-需要将单个字符映射成多大维度的向量

其实就是一个训练好的m*n大小的矩阵,根据输入词对应的序号取出对应行的向量

详细可看https://blog.csdn.net/david0611/article/details/81090371

猜你喜欢

转载自www.cnblogs.com/baitian963/p/12931142.html