如何在Torch中使用nn.LookupTable

如何使用nn.LookupTable将我的词汇索引向量转换为嵌入向量的张量?

https://github.com/torch/nn/blob/master/LookupTable.lua

网络设置如下:

self.llstm = LSTM
self.rlstm = LSTM

local modules = nn.Parallel()
  :add(nn.LookupTable(self.vocab_size,self.emb_size))
  :add(nn.Collapse(2))
  :add(self.llstm)
  :add(self.my_module)

self.params, self.grad_params = modules:getParameters

训练步骤:

input = dataset[i]
emb_input = ?
self.llstm.forward(emb_input,reverse
点赞