2024年9月12日
torch.nn.Embedding的导入与导出
摘要： 简介及导入转自：torch.nn.Embedding使用 在RNN模型的训练过程中，需要用到词嵌入，使用torch.nn.Embedding可以快速的完成：只需要初始化torch.nn.Embedding(n,m)即可（n是单词总数，m是词向量的维度）(n是嵌入字典的大小，m是嵌入向量的维度。)。 阅读全文
posted @ 2024-09-12 20:30 kingwzun 阅读(18) 评论(0) 推荐(0) 编辑