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

seqmodel with raw arrays

Browse Source
This commit is contained in:
Mike J Innes 2017-04-19 15:13:17 +01:00
parent f8a3b02c1d
commit 42a8117704
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/model.jl
View File

@ -151,4 +151,5 @@ end
splitseq(xs) = unstack(rawbatch(xs), 2)
joinseq(xs) = rebatchseq(stack(xs, 2))
(m::SeqModel)(x::BatchSeq) = joinseq(m.model((splitseq(x)...,)))
(m::SeqModel)(x::AbstractArray) = stack(m((unstack(x, 2)...,)), 2)
(m::SeqModel)(x::BatchSeq) = rebatchseq(m(rawbatch(x)))