removed zeros fix

2018-07-17 17:13:55 +02:00 · 2018-07-17 17:13:55 +02:00 · 8ab209126d
commit 8ab209126d
parent e98538673a
1 changed files with 1 additions and 1 deletions
--- a/src/layers/basic.jl
+++ b/src/layers/basic.jl
@ -101,7 +101,7 @@ struct Diagonal{T}
  β::T
 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)))

@treelike Diagonal