nyyngh的博客

torch.nn.Embedding(num_embeddings,embedding_dim,.....其他参数)生成嵌入：第一个参数是需要生成多少个嵌入，第二个参数是生成嵌入的维度embedding的取值是正态分布N（０，１）取值，好像不太对代码验证import torchimport torch.nn as nnx=torch.LongTensor([1,2,3,5,2,3])print(x)embeddings = nn.Embedding(6,2)print(embedding