pool padding

FluxML · Dec 18, 2017 · 980c768 · 980c768
1 parent 56bb473
commit 980c768
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/conv.jl b/src/conv.jl
@@ -228,5 +228,5 @@ for (T,S) in ((Float32,32), (Float64,64)); @eval begin
     end
 end;end
 
-maxpool2d(x, k) = pool(x; window = k, mode = 0)
-avgpool2d(x, k) = pool(x; window = k, mode = 1)
+maxpool2d(x, k; pad = 0) = pool(x; window = k, padding = pad, mode = 0)
+avgpool2d(x, k; pad = 0) = pool(x; window = k, padding = pad, mode = 1)