Im2col 算法
Im2col 算法
2024年7月6日创建
505
547
在 AI 框架早期，Caffe 使用 Im2col 方法将三维张量转换为二维 矩阵，
从而充分利用已经优化好的 GEMM 库来为各个平台加速卷积计算
Img2col 算法主要包含两个步骤
1.
Im2col 将输入矩阵展开一个大矩阵，矩阵每一列表示卷积核需要的一个输入数据，
2.
使用上面转换的矩阵进行 Matmul 运算，得到的数据就是最终卷积计算的结果。
3.
Col2Im 将结果转换为三维矩阵输出
卷积默认采用数据排布方式为 NHWC，
输入维度为4维(N,IH,IW,IC)，卷积核维度为(OC, KH, K W , IC)，输出维度为(N, OH, O W , OC)。
conv 转换为 gemm 计算
附件不支持打印
加载失败，