为什么转换为im2col的速度很快?

时间:2020-02-29 17:11:34

标签: deep-learning conv-neural-network caffe

原始输入激活通过im2col进行内核转换,以改善内存访问模式。但是,当我们将原始矩阵转换为im2col矩阵时,我们还将访问相同的原始内存模式。那么,为什么im2col操作本身并不慢?

1 个答案:

答案 0 :(得分:0)

使用im2col的主要原因是输入和内核可以表示为两个大矩阵,并且卷积可以在单个矩阵乘法中完成。因为矩阵乘法可以是 并行化很好。

仅仅是内存访问不是问题,正如您所说的,im2col必须像简单的卷积操作一样访问原始张量。