TastyPancakes
/
Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix conv2d shape inference

This commit is contained in:
Mike J Innes 2016-10-10 23:20:40 +01:00
parent 4961bf72af
commit 438dc9d40a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/layers/shims.jl
View File

@ -1,7 +1,7 @@
export Conv2D, MaxPool
type Conv2D <: Model
filter::Param{Array{Float32,4}} # [height, width, outchans, inchans]
filter::Param{Array{Float32,4}} # [height, width, inchans, outchans]
stride::Dims{2}
end
@ -9,7 +9,7 @@ Conv2D(size; in = 1, out = 1, stride = (1,1), init = initn) =
Conv2D(param(initn(size..., in, out)), stride)
shape(c::Conv2D, in::Dims{2}) =
(map(i -> (in[i]-size(c.filter,i))÷c.stride[i]+1, (1,2))..., size(c.filter, 3))
(map(i -> (in[i]-size(c.filter,i))÷c.stride[i]+1, (1,2))..., size(c.filter, 4))
shape(c::Conv2D, in::Dims{3}) =
shape(c, (in[1],in[2]))