【发布时间】:2017-08-03 13:53:06
【问题描述】:
我对 torch7 中的 nn 包感到困惑,尤其是它的表格层。如何使用它的 ConcatTable、CAddTable 或任何其他方法在下面创建网络?
我的网络有两个分支,一个包含 3 个卷积层,另一个只有一层。我想对我最后两个卷积层的输出求和(第 4 卷积和第 5 卷积的输出求和),应该如何使用 nn 包编写我的火炬代码。
【问题讨论】:
标签: deep-learning torch resnet
我对 torch7 中的 nn 包感到困惑,尤其是它的表格层。如何使用它的 ConcatTable、CAddTable 或任何其他方法在下面创建网络?
我的网络有两个分支,一个包含 3 个卷积层,另一个只有一层。我想对我最后两个卷积层的输出求和(第 4 卷积和第 5 卷积的输出求和),应该如何使用 nn 包编写我的火炬代码。
【问题讨论】:
标签: deep-learning torch resnet
假设您的分支正确实现:
local net = nn.Sequential()
:add(conv1)
:add(nn.ConcatTable()
:add(branch1)
:add(branch2))
:add(nn.CAddTable())
nn.CAddTable 不能与nn.Sum 混淆。第一个接收张量表并返回所有张量的总和,而第二个接收单个张量并沿指定维度计算其元素的总和。
【讨论】: