Pytorch中nn.Embedding的原理及使用

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

 

全部评论

相关推荐

2025-11-07 11:05
已编辑
腾讯_csig_软件开发(实习员工)
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务