Embedding是torch.nn下的一个类,主要负责进行数据的嵌入,如
import torch
embedding = torch.nn.Embedding(10, 10)
前面的10的最大的嵌入的值,后面的10是嵌入的维度
即输入嵌入值的范围为[0-9]
通过 word embedding,就可以将自然语言所表示的单词或短语转换为计算机能够理解的由实数构成的向量或矩阵形式
嵌入的对象必须是张量,如torch.LongTensor([2])作为被嵌入的数字,得到的输出为:
tensor([[-0.0980, 0.7833, -0.3387, -0.5940, 1.3079, 1.0000, -0.5075, -0.6461,
0.7159, 0.4094]], grad_fn=)
将数字2嵌入到了一个10维的空间中