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