在 Torch Lua 代码中找不到 addmm 函数的定义
2018-5-28 12:51:14
收藏:0
阅读:89
评论:1
我正在理解在 Torch Lua 中实现的神经网络。在通过线性层的反向传递期间,它调用了一个名为 Linear:updateGradInput 的函数( https://github.com/torch/nn/blob/master/Linear.lua#L75 )
function Linear:updateGradInput(input, gradOutput)
if self.gradInput then
local nElement = self.gradInput:nElement()
self.gradInput:resizeAs(input)
if self.gradInput:nElement() ~= nElement then
self.gradInput:zero()
end
if input:dim() == 1 then
self.gradInput:addmv(0, 1, self.weight:t(), gradOutput)
elseif input:dim() == 2 then
self.gradInput:addmm(0, 1, gradOutput, self.weight)
end
return self.gradInput
end
end
在该函数中,通过调用一个名为 addmm 的函数执行了一个基本的矩阵乘法操作(https://github.com/torch/nn/blob/master/Linear.lua#L86 )。我找不到这个 addmm 函数的定义。
在 TH 库中定义了一个 addmm 函数( https://github.com/torch/torch7/blob/master/lib/TH/generic/THTensorMath.c#L1282),但是我不确定 Lua 代码与 C 中的这段代码有何关联。
点赞
评论区的留言会收到邮件通知哦~
推荐文章
- 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 代码?

刚刚发现了 Lua 代码和 C 代码之间的联系。在 Lua 代码中对 addmm 的调用引导到这个函数(https://github.com/torch/torch7/blob/master/TensorMath.lua#L487-L510),然后这个函数调用在 C Torch Library 中定义的 addmm 函数,它在这里定义(https://github.com/torch/torch7/blob/master/lib/TH/generic/THTensorMath.c#L1282)。
这很棘手,因为 Lua 通过字符串构建对 C 函数的调用。