0
点赞
收藏
分享

微信扫一扫

nn.Embedding 学习笔记

迪莉娅1979 2022-02-09 阅读 79

文章目录

1. 说明

torch.nn.Embedding官网链接

  • 作用:一个简单的查找表,存储固定字典和大小的嵌入。这个模块通常用于存储单词嵌入并使用索引检索它们。模块的输入是一个索引列表,输出是相应的单词嵌入。
  • 函数:
class torch.nn.Embedding(num_embeddings, embedding_dim, padding_idx=None, max_norm=None, norm_type=2.0, scale_grad_by_freq=False, sparse=False, _weight=None, device=None, dtype=None)

num_embeddings:嵌入字典的大小

举报

相关推荐

0 条评论