卷积操作乘法次数计算:2x10x10 Tensor 与 10x2x3x3 Kernel
对于 2x10x10 的 Tensor 和 10x2x3x3 的 Kernel 进行卷积操作,需要进行的乘法次数可以通过如下计算得到:
每个输出元素需要进行 3x3=9 次乘法操作,其中有 10x10 个输出元素。所以总共需要进行的乘法次数为:9x10x10=900 次。
如果 Kernel 中的 3x3 系数可以横竖分解,则可以将每个输出元素的计算分解为两个部分:
首先,对于每个输入通道,将 3x3 系数分解为一个 3x1 和一个 1x3 的乘法。这样,每个输入通道需要进行的乘法次数为:3x1+1x3=6 次。
然后,对于每个输出元素,需要对 10 个输入通道进行计算。所以,每个输出元素需要进行的乘法次数为:10x6=60 次。
最后,总共有 10x10 个输出元素,所以总共需要进行的乘法次数为:60x10x10=6000 次。
因此,如果 Kernel 中的 3x3 系数可以横竖分解,则需要进行 6000 次乘法。
原文地址: https://www.cveoy.top/t/topic/qc91 著作权归作者所有。请勿转载和采集!