如何在Torch中使用NN LookupTable

时间:2016-05-09 22:17:39

标签: lua neural-network torch

如何使用nn.LookupTable将我的词汇索引向量转换为嵌入向量的张量?
https://github.com/torch/nn/blob/master/LookupTable.lua

网络设置为:

Box<X>

在火车一步:

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

0 个答案:

没有答案