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

fix nested mapt

Browse Source
This commit is contained in:
Mike J Innes 2017-03-20 23:19:03 +00:00
parent eadf96605c
commit a5bd72753e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/dims/batching.jl
View File

@ -32,7 +32,7 @@ convertel(T::Type, xs::Batch) =
# TODO: remove this in favour of full batching semantics # TODO: remove this in favour of full batching semantics
mapt(f, x) = f(x) mapt(f, x) = f(x)
mapt(f, xs::Tuple) = map(f, xs) mapt(f, xs::Tuple) = map(x -> mapt(f, x), xs)
batchone(x) = Batch((x,)) batchone(x) = Batch((x,))
batchone(x::Batch) = x batchone(x::Batch) = x