初始化嵌入 nn.Embedding
torch.nn.Embedding(num_embeddings,embedding_dim,.....其他参数)
生成嵌入:第一个参数是需要生成多少个嵌入,第二个参数是生成嵌入的维度
embedding的取值是正态分布N(0,1)取值,好像不太对
代码验证
import torch
import torch.nn as nn
x=torch.LongTensor([1,2,3,5,2,3])
print(x)
embeddings = nn.Embedding(6,2)
print(embedding
原创
2021-04-26 20:04:11 ·
1867 阅读 ·
0 评论