don't rebatch batches

2016-10-30 00:20:15 +01:00 · 2016-10-30 00:20:15 +01:00 · 605e3a9363
commit 605e3a9363
parent 06e37081d3
1 changed files with 1 additions and 0 deletions
--- a/src/backend/tensorflow/model.jl
+++ b/src/backend/tensorflow/model.jl
@ -20,6 +20,7 @@ function tf(model)
 end

 batchone(x) = Batch((x,))
+batchone(x::Batch) = x

 function batch(xs)
  dims = ndims(xs)-1