gpu fix

FluxML · Feb 17, 2018 · 989adcd · 989adcd
1 parent 1151198
commit 989adcd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/layers/conv.jl b/src/layers/conv.jl
@@ -29,7 +29,7 @@ Conv2D(k::NTuple{2,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity;
 Flux.treelike(Conv2D)
 
 function (c::Conv2D)(x)
-  σ, b = c.σ, reshape(c.bias, 1, 1, :)
+  σ, b = c.σ, reshape(c.bias, 1, 1, :, 1)
   σ.(conv2d(x, c.weight, stride = c.stride, padding = c.pad) .+ b)
 end