From 989adcdc7ddc62e63ab036d9781d36e0afba521c Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Sat, 17 Feb 2018 12:41:53 +0000 Subject: [PATCH] gpu fix --- src/layers/conv.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/conv.jl b/src/layers/conv.jl index 4578b75b5f..f852e781e8 100644 --- 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