Pytorch学习-Embedding
官方文档的解释:
embedding方法为一个存储了固定字典和大小的简单的查找表
通常利用indices去检索存储的word embedding,输入是一系列indices,输出对应的word embeddings
from_pretrained method
根据输入的2d FloatTensor 格式的embeddings参数 生成embedding实例,其中embeddings参数第一维度为词典大小num_embeddings , 第二维度为 词典中每个词的维度embedding_dim
具体例子
原创
2021-04-11 06:29:36 ·
448 阅读 ·
0 评论