pytorch对item进行嵌入nn.embedding()函数

nn.embedding()

用于将多个item(例如10000个item)用固定长度的向量表示,如果想要表示这1w个item,正常来说就是用onehot形式,维度为1w,太大,因此可以用nn.embedding将其嵌入为低维度的vector,内部原理是一个随机初始化的矩阵

embedding = nn.Embedding(input_dim, embedding_dim)

此时就得到了一个(input_dim,embedding_dim)的矩阵,我们每输入一个index(item的第index个,范围是[0,input_dim-1]),就可以从该矩阵中得到相应的embedding_dim维度的一个向量。

注意:输入必须是LongTensor,FloatTensor须通过**tensor.long()**方法转成LongTensor。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值