pytorch实现
embedding = embedding / embedding.norm(dim=1, keepdim=True)
wdvec = embedding[wdid, :]
dis = 1 - (wdvec *embedding).sum(dim=1)
idx = dis.argsort()
先用第二范式对embedding矩阵的每一行取模，再向量成矩阵，最后对矩阵排序得到最优解
余弦相似度实现
pytorch实现
embedding = embedding / embedding.norm(dim=1, keepdim=True)
wdvec = embedding[wdid, :]
dis = 1 - (wdvec *embedding).sum(dim=1)
idx = dis.argsort()
