[torch]nngraph parameters of the node

https://github.com/torch/nngraph/issues/131

require 'nn'
require 'nngraph'
require 'rnn'

function make_net()
        local mlp = nn.Sequential();  -- make a multi-layer perceptron
        mlp:add(nn.MaskZero(nn.Linear(100, 5),1))
        mlp:add(nn.MaskZero(nn.Tanh(),1))
        mlp:add(nn.MaskZero(nn.Linear(5, 10),1))
        h1 = nn.Identity()()
        h2 = mlp(h1):annotate{name ='mlp_layer', description = 'mlp layer'}
        return(nn.gModule({h1}, {h2}))
end

function main()
        local nnmodule =  make_net()
        x = torch.rand(20,100)

        print(nnmodule.forwardnodes)
        for _, node in ipairs(nnmodule.forwardnodes) do
                if node.data.annotations.name == 'mlp_layer' then
                        --node.data.module.weight:fill(1.0)
                        for i = 1,3 do
                                print(node.data.module.modules[i])
                                print(node.data.module.modules[i].modules)
                        end
                end
        end

        nnmodule:updateOutput(x)
end
main()
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值