From f6097d58d6416a842509c915c5d07da9e28dc4d6 Mon Sep 17 00:00:00 2001
From: tejank10 <tejank10@gmail.com>
Date: Sun, 15 Apr 2018 12:15:41 +0530
Subject: [PATCH] Scalar pad/stride for Conv constructor

---
 src/layers/conv.jl | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/layers/conv.jl b/src/layers/conv.jl
index 994648c2..e8054829 100644
--- a/src/layers/conv.jl
+++ b/src/layers/conv.jl
@@ -24,11 +24,16 @@ Conv(w::AbstractArray{T}, b::AbstractVector{T}, σ = identity;
        stride = 1, pad = 0) where T =
   Conv(σ, w, b, stride, pad)
 
+#Conv(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity; init = initn,
+#     stride::NTuple{N,Integer} = map(_->1,k),
+#     pad::NTuple{N,Integer} = map(_->0,k)) where N =
+#  Conv(param(init(k..., ch...)), param(zeros(ch[2])), σ,
+#       stride = stride, pad = pad)
+
 Conv(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity; init = initn,
-     stride::NTuple{N,Integer} = map(_->1,k),
-     pad::NTuple{N,Integer} = map(_->0,k)) where N =
+     stride::Integer = 1, pad::Integer = 0) where N =
   Conv(param(init(k..., ch...)), param(zeros(ch[2])), σ,
-       stride = stride, pad = pad)
+       stride = map(_->stride,k), pad = map(_->pad,k))
 
 Flux.treelike(Conv)