im2col + gemm建模

Question

我试图通过卷积神经网络层来理解im2col + gemm。到目前为止，这是我的理解：

因此，在conv2d层中，将有一个输入图像（带有通道，高度和宽度）和一个过滤器（在这个示例中，带有通道-与输入图像相同，并且高度/宽度相等）。将有多个过滤器。

我相信在使用im2col时，我走在正确的轨道上，但是有些错误。对于滤波器和输入k k c，分别使用宽度和高度。另一个维度是我不确定的地方。对于过滤器，我相信它是过滤器的数量-这也将是输出高度的尺寸。对于输入，我不确定如何计算宽度。我认为这取决于步幅以及是否有任何初始填充。

有什么想法吗？