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

added expand in conv constructor

Browse Source
This commit is contained in:
tejank10 2018-04-16 00:59:11 +05:30
parent b080f5c82e
commit 2f5473d435
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/layers/conv.jl
View File

@ -24,8 +24,9 @@ struct Conv{N,F,A,V}
end end
Conv(w::AbstractArray{T}, b::AbstractVector{T}, σ = identity; Conv(w::AbstractArray{T}, b::AbstractVector{T}, σ = identity;
stride = 1, pad = 0) where T = stride = 1, pad = 0) where T =
Conv(σ, w, b, stride, pad) Conv(σ, w, b, expand(Val{ndims(w) - 2}, stride),
pad = expand(Val{ndims(w) - 2}, pad))
Conv(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity; init = initn, Conv(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity; init = initn,
stride = 1, pad = 0) where N = stride = 1, pad = 0) where N =