torch再次入门之重出江湖torch.nn.Embedding

最新推荐文章于 2024-07-04 12:17:17 发布

小李飞刀李寻欢

最新推荐文章于 2024-07-04 12:17:17 发布

阅读量868

点赞数 2

分类专栏： python 文章标签： torch Embedding

本文链接：https://blog.csdn.net/SPESEG/article/details/108215199

版权

python 专栏收录该内容

209 篇文章 69 订阅 ¥9.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了torch.nn.Embedding的功能和使用，它是一个用于存储固定大小词典的查找表，输入为索引列表，输出为对应的词嵌入。讨论了其参数如num_embeddings和embedding_dim，并举例说明了不同形状输入的处理方式，同时提到了预训练词嵌入的使用和优化器的支持情况。

摘要由CSDN通过智能技术生成

嗨各位大佬好，之前初步了解了torch的东西，但不够深入，这次有机会再次学习。

源于一个推荐方面的Chorus，自搜可得。

torch.nn.Embedding

一个简单的存储固定字典和size的查找表，输入是索引的list，输出是相应的词嵌入。

 Args:
 |      num_embeddings (int): size of the dictionary of embeddings
 |      embedding_dim (int): the size of each embedding vector
 |      padding_idx (int, optional): If given, pads the output with the embedding vector at :attr:`padding_idx`
 |                                       (initialized to zeros) whenever it encounters the index.
 |      max_norm (float, optional): If given, each embedding vector with norm larg