我尝试解决这个问题已经有一段时间了,仍然没有找到解决方案,希望有人能帮我解决。我有一个如下所示的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表层在网络内部将AAABBB表转换为ABABAB嵌套表?
回答:
似乎dpnn包包含了一个容器,它可以完成这个任务。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})