pytorch - 如何将word2vec保存为单词和向量pytorch

Question

这是保存功能

def write_embedding_to_file(self,filename):
        for i in self.embeddings.parameters():
            weights = i.data.numpy()
        np.save(filename,weights)

我只从那里得到权重向量 如何将它们保存到 dict{word:vectors} 我正在使用 pytorch