深度学习Torch源代码解析卷积层细节
发布时间
阅读量:
阅读量
以向前传播过程为研究对象,本文对Torch框架中nn模块内SpatialConvolution函数的具体实现机制进行了深入探讨。在解析相关源代码的过程中,不仅提供了对应的GitHub代码仓库链接,还标注了该文件在系统安装完成后所处的存储路径。
初始化
构建一个卷积层需提供以下输入参数
nInputPlane\nOutputPlane -- 输入\输出通道数,M\N
kW\kH -- 核尺寸,K
dW\dH -- 步长
padW\padH -- 补边
卷积层中的关键参数
weight -- 卷积核权重,N*M*K*K
bias -- 卷积核偏置,N
gradWeight -- 权重导数,N*M*K*K
gradBia -- 偏置导数,N
全部评论 (0)
还没有任何评论哟~
