Merge pull request #93 from FluxML/sf/im2col_testing

Fix typo in plane size calculation
FluxML · Feb 9, 2019 · d07ac0b · d07ac0b
2 parents 42d9f64 + 0163bec
commit d07ac0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/impl/conv.jl b/src/impl/conv.jl
@@ -342,7 +342,7 @@ function conv2d!(y::AbstractArray{T,4}, x::AbstractArray{T,4}, w::AbstractArray{
     Ww, Hw = kernel_size(cdims)
     Wy, Hy = output_size(cdims)
     Cx = img_channels(cdims)
-    M, N, K, Y = Wy*Hy, size(y,4), Ww*Hw*Cx, Wy*Hy*size(y, 4)
+    M, N, K, Y = Wy*Hy, size(y,3), prod(size(w)[1:3]), prod(size(y)[1:3])
 
     x2 = similar(x, im2col_dims(w, y))
     @inbounds for n in 1:size(x,4)