From c6e51f5cc2c32c844e89b79a4d4cb39d7802bd3f Mon Sep 17 00:00:00 2001
From: Manjunath Bhat <manjunathbhat9920@gmail.com>
Date: Thu, 7 Mar 2019 23:42:38 +0530
Subject: [PATCH] Made lambda and alpha of eltype(x)

---
 src/layers/normalise.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/layers/normalise.jl b/src/layers/normalise.jl
index 7aeaeade..5fd93e9d 100644
--- a/src/layers/normalise.jl
+++ b/src/layers/normalise.jl
@@ -61,8 +61,8 @@ end
 
 function (a::AlphaDropout)(x)
   a.active || return x
-  λ = 1.0507009873554804934193349852946
-  α = 1.6732632423543772848170429916717
+  λ = eltype(x)(1.0507009873554804934193349852946)
+  α = eltype(x)(1.6732632423543772848170429916717)
   α1 = eltype(x)(-λ*α)
   noise = randn(eltype(x), size(x))
   x = @. x*(noise > (1 - a.p)) + α1 * (noise <= (1 - a.p))