將張量表的AAABBB錶轉換爲火炬中張量的嵌套ABABAB表

Question

我試圖解決這個問題很長一段時間，仍然不能拿出解決方案，也許有人可以幫我解決這個問題。我有以下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   
    } 
} 

如何我可以將AAABBB表格更改爲ABABAB嵌套表內網絡使用Torch table layers？

Answer 1

看來，dpnn包包含一個Container完全是。 ZipTable將表格表格拉入表格中。

下面是如何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})