im2col + gemm建模

时间:2019-07-17 00:11:36

标签: machine-learning conv-neural-network matrix-multiplication

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

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

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

有什么想法吗?

0 个答案:

没有答案