参数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)