Caffe 中convolution的计算

Caffe 中convolution的计算_第1张图片Caffe 中convolution的计算_第2张图片Caffe 中convolution的计算_第3张图片Caffe 中convolution的计算_第4张图片

最后一页没画,但是基本上就是Filter Matrix乘以Feature Matrix的转置,得到输出矩阵Cout x (H x W),就可以解释为输出的三维Blob(Cout x H x W)。

Caffe里用的是CHW的顺序,有些library也会用HWC的顺序(比如说CuDNN是两个都支持的),这个在数学上其实差别不是很大,还是一样的意思。

顺便广告一下我吐槽Caffe里面卷积算法的链接:Convolution in Caffe: a memo · Yangqing/caffe Wiki · GitHub

作者:贾扬清
链接:https://www.zhihu.com/question/28385679/answer/44297845
来源:知乎
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

你可能感兴趣的:(机器学习)