转：torch.nn.Embedding函数用法图解

【python函数】torch.nn.Embedding函数用法图解-CSDN博客

?

1 2 3 4 5 6 7 8 9 10 11

import torch import torch.nn as nn   embedding = nn.Embedding(10, 3)   # 10表示num_embeddings, 3表示embedding_dim。用标准正态分布进行权重元素的初始化。这些权重是learnable的。 x = torch.LongTensor([[1, 2, 4, 5], [4, 3, 2, 9]])   # 从10个embeddings中取出第1个、第2个、...第9个   y = embedding(x)   print('权重:\n', embedding.weight) print('输出:') print(y)