TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

don't rebatch batches

Browse Source
This commit is contained in:
Mike J Innes 2016-10-30 00:20:15 +01:00
parent 06e37081d3
commit 605e3a9363
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/backend/tensorflow/model.jl
View File

@ -20,6 +20,7 @@ function tf(model)
end end
batchone(x) = Batch((x,)) batchone(x) = Batch((x,))
batchone(x::Batch) = x
function batch(xs) function batch(xs)
dims = ndims(xs)-1 dims = ndims(xs)-1