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

removed zeros fix

Browse Source
This commit is contained in:
pevnak 2018-07-17 17:13:55 +02:00 committed by Mike J Innes
parent e98538673a
commit 8ab209126d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/basic.jl
View File

@ -101,7 +101,7 @@ struct Diagonal{T}
β::T β::T
end end
Diagonal(in::Integer; initα = ones, initβ = (x) -> similar(x) .= 0) = Diagonal(in::Integer; initα = ones, initβ = zeros) =
Diagonal(param(initα(in)), param(initβ(in))) Diagonal(param(initα(in)), param(initβ(in)))
@treelike Diagonal @treelike Diagonal