如何在torch中向图形模块添加附加层
2016-10-31 13:24:7
收藏：0
阅读：74
评论：1
如何在torch的nngraph包中向图形模块(gModule)添加新节点？我尝试使用add函数，这将节点添加到gModules对象中的module插槽中。然而，输出仍然来自于先前的最后一个节点。
简化代码：
require" nn"
require" nngraph"
--构建gModule的功能
function buildModule(input_size,hidden_size)
local x = nn.Identity()()
local out = x-nn.Linear(input_size,hidden_size)-nn.Tanh()
return nn.gModule({x},{out})
end
network = buildModule(5,3)
--要添加的附加层
l2 = nn.Linear(3,10)
network:add(l2)
--预期尺寸为10的张量，但获得了一个尺寸为3的张量
print(network:forward(torch.randn(5)))
点赞
评论区的留言会收到邮件通知哦~
作者:
用户1074060
推荐文章
- Lua 虚拟机加密load(string.dump(function)) 后执行失败问题如何解决
- 我想创建一个 Nginx 规则，禁止访问
- 如何将两个不同的lua文件合成一个 东西有点长 大佬请耐心看完 我是小白研究几天了都没搞定
- 如何在roblox studio中1：1导入真实世界的地形？
- 求解，lua_resume的第二次调用继续执行协程问题。
- 【上海普陀区】内向猫网络招募【Skynet游戏框架Lua后端程序员】
- SF爱好求教：如何用lua实现游戏内调用数据库函数实现账号密码注册？
- Lua实现网站后台开发
- LUA错误显式返回，社区常见的规约是怎么样的
- lua5.3下载库失败
- 请问如何实现文本框内容和某个网页搜索框内容连接，并把网页输出来的结果反馈到另外一个文本框上
- lua lanes多线程使用
- 一个kv数据库
- openresty 有没有比较轻量的 docker 镜像
- 想问一下，有大佬用过luacurl吗
- 在Lua执行过程中使用Load函数出现问题
- 为什么 neovim 里没有显示一些特殊字符?
- Lua比较两个表的值（不考虑键的顺序）
- 有个lua简单的项目，外包，有意者加微信 liuheng600456详谈，最好在成都
- 如何在 Visual Studio 2022 中运行 Lua 代码？
gModule实际上不应该被改变。它支持:add的事实实际上是它作为 nn.Container 的子类的副作用，而不是设计决策。通常情况下，一旦创建了gModule，就不应该修改它的内部结构，因为你需要修改一些内部属性才能使所有工作正常。相反，如果你想要“在顶部”添加一些东西，只需定义一个新的容器，将前一个容器作为输入。-- 构建一个 gModule 的函数 function buildModule(input_size,hidden_size) local x = nn.Identity()() -- 输入层 local out = x - nn.Linear(input_size,hidden_size) - nn.Tanh() -- 隐藏层和输出层 return nn.gModule({x},{out}) -- 返回 gModule end network = buildModule(5,3) -- 构建网络 new_network = nn.Sequential() -- 定义新的容器 new_network:add(network) -- 添加网络 new_network:add(nn.Linear(3,10)) -- 在顶部添加一些东西