跳转至

17.2 卷积前向计算代码实现

17.2 卷积前向计算代码实现⚓︎

以下为本小节目录,详情请参阅《智能之门》正版图书,高等教育出版社。

17.2.1 卷积核的实现⚓︎

17.2.2 卷积前向运算的实现 - 方法1⚓︎

17.2.3 卷积前向运算的实现 - 方法2⚓︎

17.2.4 卷积前向运算的实现 - 方法3⚓︎

代码位置⚓︎

ch17, Level2

其中,Level2_Numba_Test.py是测试numba库的性能,Level2_Img2Col_Test是比较numba方法和im2col方法的前向计算性能。

在Level2的主过程中有4个函数:

  • test_2d_conv,理解2维下im2col的工作原理
  • understand_4d_im2col,理解4维下im2col的工作原理
  • test_4d_im2col,比较两种方法的结果,从而验证正确性
  • test_performance,比较两种方法的性能