【问题标题】:How can I use table layers in torch to create my own network?如何在 Torch 中使用表层来创建自己的网络?
【发布时间】:2017-08-03 13:53:06
【问题描述】:

我对 torch7 中的 nn 包感到困惑,尤其是它的表格层。如何使用它的 ConcatTable、CAddTable 或任何其他方法在下面创建网络?

我的网络有两个分支,一个包含 3 个卷积层,另一个只有一层。我想对我最后两个卷积层的输出求和(第 4 卷积和第 5 卷积的输出求和),应该如何使用 nn 包编写我的火炬代码。

【问题讨论】:

    标签: deep-learning torch resnet


    【解决方案1】:

    假设您的分支正确实现:

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

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

    【讨论】:

    • 你能看看this这个帖子和这个帖子有点相似,帮助我克服它吗?
    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2016-02-15
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2015-09-09
    • 1970-01-01
    • 2018-01-09
    相关资源
    最近更新 更多