参数1:vocab的大小

参数2:d_model

参数3(padding_idx):可选参数,原张量中置为idx的词,都被映射成一个长为d_model的零向量

例1

embedding=torch.nn.Embedding(10,3)
input = torch.LongTensor([[1,2,4,5],[4,3,2,9]])
embedding(input)

例2

embedding=torch.nn.Embedding(10,3)
input = torch.LongTensor([[1,2,4,5],[4,3,2,9]],padding_idx=1)
embedding(input)