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

still need batchone

Browse Source
This commit is contained in:
Mike J Innes 2017-04-19 13:26:37 +01:00
parent 5357b1e9f9
commit f6334e7e96
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/dims/batching.jl
View File

@ -27,3 +27,6 @@ end
convertel(T::Type, xs::Batch) = convertel(T::Type, xs::Batch) =
eltype(eltype(xs)) isa T ? xs : eltype(eltype(xs)) isa T ? xs :
Batch(map(x->convertel(T, x), xs)) Batch(map(x->convertel(T, x), xs))
batchone(x) = Batch((x,))
batchone(x::Batch) = x