卷积计算次数:2x10x10 Tensor 与 10x2x3x3 Kernel 的乘法分析
对于一个 2x10x10 的 Tensor 和一个 10x2x3x3 的 Kernel 进行卷积操作,每个输出值需要进行多少次乘法,取决于 Kernel 的大小。
对于一个 2x10x10 的 Tensor,它的大小是 2x10x10 = 200 个元素。
对于一个 10x2x3x3 的 Kernel,它的大小是 10x2x3x3 = 180 个元素。
在进行卷积操作时,每个输出值都是通过将 Kernel 与 Tensor 的对应部分进行元素相乘,然后求和得到的。所以每个输出值需要进行 180 次乘法操作。
因此,对于整个 2x10x10 的 Tensor 和 10x2x3x3 的 Kernel 进行卷积操作,总共需要进行 200x180 = 36000 次乘法。
如果 Kernel 中的 3×3 系数可以横竖分解,即可以写成一个 1x3 的行向量和一个 3x1 的列向量的乘积,那么每个输出值只需要进行 3 次乘法操作。
因此,在这种情况下,对于整个 2x10x10 的 Tensor 和 10x2x3x3 的 Kernel 进行卷积操作,总共只需要进行 200x3 = 600 次乘法。
原文地址: https://www.cveoy.top/t/topic/qc9z 著作权归作者所有。请勿转载和采集!