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

add N

Browse Source
This commit is contained in:
Dhairya Gandhi 2019-10-06 04:55:33 +05:30
parent 2ae3ad3b31
commit 214f71f492
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/layers/conv.jl
View File

@ -50,7 +50,8 @@ function Conv(w::AbstractArray{T,N}, b::Union{Number, AbstractVector{T}}, σ = i
return Conv(σ, w, b, stride, pad, dilation) return Conv(σ, w, b, stride, pad, dilation)
end end
convweight(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}; init = glorot_uniform) = init(k..., ch...) convweight(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer};
init = glorot_uniform) where N = init(k..., ch...)
const convbias = zeros const convbias = zeros
function Conv(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity; function Conv(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity;