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

this might be a bit dirty

Browse Source
This commit is contained in:
Mike J Innes 2017-06-02 15:22:23 +01:00
parent 215e997540
commit 4698cc0156
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/dims/seq.jl
View File

@ -37,6 +37,7 @@ end
runseq(f, xs::Tuple...) = f(xs...) runseq(f, xs::Tuple...) = f(xs...)
runseq(f, xs::AbstractArray...) = stack(f(map(x -> (unstack(x,2)...,), xs)...), 2) runseq(f, xs::AbstractArray...) = stack(f(map(x -> (unstack(x,2)...,), xs)...), 2)
runseq(f, xs::BatchSeq...) = rebatchseq(runseq(f, rawbatch.(xs)...)) runseq(f, xs::BatchSeq...) = rebatchseq(runseq(f, rawbatch.(xs)...))
runseq(f, xs) = runseq(f, (xs...,))
function (m::SeqModel)(x) function (m::SeqModel)(x)
runseq(x) do x runseq(x) do x