Merge pull request #955 from yangfei963158659/develop

repair bug of im2col

Merge pull request #955 from yangfei963158659/develop
repair bug of im2col
9da43434 · smilejames · GitHub · 58a9e04b · eb966ff5 · 9da43434
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

src/operators/math/im2col.cpp src/operators/math/im2col.cpp +1 -1

未找到文件。
--- a/src/operators/math/im2col.cpp
+++ b/src/operators/math/im2col.cpp
@@ -74,7 +74,7 @@ class Im2ColFunctor<ColFormat::kCFO, CPU, T> {
    const int isize = im_height;
    bool pad1 = padding[0] > 0;
    bool pad2 =
-        (pad1 &&
+        (pad1 && padding[1] &&
         (((isize - 2 * padding[0] + filter_height) % stride[0] == 0) ? 1 : 0));
    int fill = isize % 2;
    if (stride[0] == 1 && filter_height == 3 && pad1 && pad2 &&