From edfb0211e665814781e4e94feb9d86bc4417c95b Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Wed, 19 Apr 2017 17:18:40 +0100
Subject: [PATCH] better for nested batches

---
 src/utils.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/utils.jl b/src/utils.jl
index e9d0eddc..5f1e2807 100644
--- a/src/utils.jl
+++ b/src/utils.jl
@@ -5,7 +5,7 @@ const AArray = AbstractArray
 initn(dims...) = randn(dims...)/100
 
 tobatch(xs::Batch) = rawbatch(xs)
-tobatch(xs) = unsqueeze(xs)
+tobatch(xs) = tobatch(batchone(xs))
 
 function train!(m, train, test = []; epoch = 1, η = 0.1)
     i = 0