From f3551da5a2ed404879f7bf49e1fe746e022e7d0b Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Wed, 24 Jul 2019 11:20:39 -0400
Subject: [PATCH] dropout printing

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

diff --git a/src/layers/normalise.jl b/src/layers/normalise.jl
index 5a8bdc56..728c91df 100644
--- a/src/layers/normalise.jl
+++ b/src/layers/normalise.jl
@@ -37,6 +37,12 @@ end
 
 (a::Dropout)(x) = dropout(x, a.p; dims = a.dims)
 
+function Base.show(io::IO, d::Dropout)
+  print(io, "Dropout(", d.p)
+  d.dims != (:) && print(io, ", dims = $(repr(d.dims))")
+  print(io, ")")
+end
+
 """
     AlphaDropout(p)
 A dropout layer. It is used in Self-Normalizing Neural Networks.