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

convertel for batch

Browse Source
This commit is contained in:
Mike J Innes 2017-01-26 23:37:06 +05:30
parent 8db503eafa
commit b79e536c13
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/dims/batching.jl
View File

@ -36,3 +36,7 @@ function rebatch(xs)
B = Array{eltype(xs),dims+1} B = Array{eltype(xs),dims+1}
Batch{T,B}(xs) Batch{T,B}(xs)
end end
convertel(T::Type, xs::Batch) =
isa(eltype(eltype(xs)), T) ? xs :
Batch(map(x->convertel(T, x), xs))