Advertisement

深度学习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)

还没有任何评论哟~