Winograd 变换

原文地址:http://ncic.ac.cn/codesign/blog/?p=2565

将一个2 X 3 的矩阵和一个3 X 1 的矩阵进行矩阵乘,我们可以用下面的公式表达:


其中:


原始的矩阵乘法需要6 次乘法,而winograd 只需要4 次。我们可以将上面的公式扩展到矩阵的表示:


其中,* 符号表示矩阵元素乘(element-wise multiplication)。对于F(2; 3) 的情况,这些矩阵如下所示:


扩展到2 维度形式以后,我们可以这样表示:


因此,我们可以在filter 维度较小情况下应用winograd 做卷积。

猜你喜欢

转载自blog.csdn.net/dengdengma520/article/details/80541598