caffe中的卷积计算

思想

将卷积运算转换为矩阵运算GMEE，利用GPU对于矩阵乘运算的高效性。

步骤

• img2col
• gemm
• col2img

示意图

Berkeley给了关于CNN优化的talk

参考文献

[1] High Performance Convolutional Neural Networks for Document Processing

[2] Convolution in Caffe: a memo · Yangqing/caffe Wiki · GitHub

[3] 在 Caffe 中如何计算卷积？ - zhihu.com