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

Scalar pad/stride for Conv constructor

Browse Source
This commit is contained in:
tejank10 2018-04-15 12:15:41 +05:30
parent 0ba5ce4601
commit f6097d58d6
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/layers/conv.jl
View File

@ -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)