将AAABBB张量表转换为Torch中嵌套的ABABAB张量表

时间:2017-03-01 16:59:56

标签: machine-learning lua torch

我试图解决这个问题很长一段时间仍然无法提出解决方案,也许有人可以帮我解决这个问题。我对神经网络有以下 AAABBB 输入表:

{
  1 : 
    {
      1 : DoubleTensor - size: 32x200
      2 : DoubleTensor - size: 32x200
      3 : DoubleTensor - size: 32x200
    }
  2 : 
    {
      1 : DoubleTensor - size: 32x54
      2 : DoubleTensor - size: 32x54
      3 : DoubleTensor - size: 32x54
    }
}

上表经过预处理,然后需要转换为嵌套 ABABAB 输入表:

{
  1 : 
    {
      1 : DoubleTensor - size: 32x200
      2 : DoubleTensor - size: 32x54
    }
  2 :
    {
      1 : DoubleTensor - size: 32x200
      2 : DoubleTensor - size: 32x54
    }
  3 :
    {
      1 : DoubleTensor - size: 32x200
      2 : DoubleTensor - size: 32x54        
    }
}

如何使用Torch table layers AAABBB 表转换为 ABABAB 嵌套表格 网络

1 个答案:

答案 0 :(得分:0)

似乎 dpnn 包中包含一个ContainerZipTable将表格表压缩到表格表中。

以下是如何将 AAABBB 表转换为 ABABAB嵌套表的工作示例。

require 'dpnn'

aaa = torch.DoubleTensor(3,32,200)
bbb = torch.DoubleTensor(3,32,54)

model = nn.Sequential()
par = nn.ParallelTable()
par:add(nn.SplitTable(1))
par:add(nn.SplitTable(1))
model:add(par)
model:add(nn.ZipTable())
model:forward({aaa,bbb})