使用火炬

Question

我现在有一个有2个输入X和Y的网络。

X连接Y然后传递给网络以获取result1。同时X会将result1作为捷径连接起来。

如果只有一个输入就很容易。

branch = nn.Sequential()
branch:add(....) --some layers
net = nn.Sequential()
net:add(nn.ConcatTable():add(nn.Identity()):add(branch))
net:add(...)

但是当涉及两个输入时，我实际上不知道该怎么做？此外，不允许使用nngraph。任何人都知道怎么做吗？

Answer 1

您可以使用表格模块，查看此页面：https://github.com/torch/nn/blob/master/doc/table.md

net = nn.Sequential()
triple = nn.ParallelTable()
duplicate = nn.ConcatTable()
duplicate:add(nn.Identity())
duplicate:add(nn.Identity())
triple:add(duplicate)
triple:add(nn.Identity())
net:add(triple)
net:add(nn.FlattenTable())
-- at this point the network transforms {X,Y} into {X,X,Y}
separate = nn.ConcatTable()
separate:add(nn.SelectTable(1))
separate:add(nn.NarrowTable(2,2))
net:add(separate)
-- now you get {X,{X,Y}}
parallel_XY = nn.ParallelTable()
parallel_XY:add(nn.Identity()) -- preserves X
parallel_XY:add(...) -- whatever you want to do from {X,Y}
net:add(parallel)
parallel_Xresult = nn.ParallelTable()
parallel_Xresult:add(...)  -- whatever you want to do from {X,result}
net:add(parallel_Xresult)

output = net:forward({X,Y})

我们的想法是从{X,Y}开始，复制X并进行操作。这显然有点复杂，nngraph应该就是这样做的。