From 8ab209126d9f8478dd7fb8ab351f37b0bab22a60 Mon Sep 17 00:00:00 2001 From: pevnak Date: Tue, 17 Jul 2018 17:13:55 +0200 Subject: [PATCH] removed zeros fix --- src/layers/basic.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/basic.jl b/src/layers/basic.jl index 8f9da6ff..d461c95c 100644 --- 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