From 15c4f38130f27d2c3bf1356c2899bf6018c47280 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Sat, 19 Aug 2017 11:00:55 +0100
Subject: [PATCH] printing

---
 src/grad/track.jl | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/grad/track.jl b/src/grad/track.jl
index 5938e806..b8125aa4 100644
--- a/src/grad/track.jl
+++ b/src/grad/track.jl
@@ -40,6 +40,17 @@ function back!(x::Var, Δ)
   back!(x.f, Δ)
 end
 
-for f in :[Base.size, Base.getindex].args
+for f in :[Base.size, Base.ndims, Base.similar].args
   @eval @inline $f(x::Var, a...) = $f(data(x), a...)
 end
+
+function Base.showarray(io::IO, X::Var, repr::Bool = true; header = true)
+  if repr
+    print(io, "Var(")
+    Base.showarray(io, data(X), true)
+    print(io, ")")
+  else
+    println(io, summary(X), ":")
+    Base.showarray(io, data(X), false, header = false)
+  end
+end