如何在火炬中使用表格层来创建自己的网络？

Question

我在torch7中遇到了关于nn包的问题，​​特别是它的表层。如何使用其ConcatTable，CAddTable或任何其他方法在下面创建网络？

这里我有两个网络分支，一个包含3个卷积层，另一个只包含一个层。我想总结我最后两个卷积层的输出（将卷积4和卷积5的输出相加），应该如何使用nn包写我的火炬代码。

Answer 1

假设您的分支机构正确实施：

local net = nn.Sequential()
    :add(conv1)
    :add(nn.ConcatTable()
        :add(branch1)
        :add(branch2))
    :add(nn.CAddTable())

nn.CAddTable不要与nn.Sum混淆。第一个接收张量表并返回所有张量的总和，而第二个接收单个张量，并计算沿指定维度的元素总和。