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

unrolled graphs only have one output

Browse Source
This commit is contained in:
Mike J Innes 2017-06-02 16:22:39 +01:00
parent 50bbb87674
commit baaf9f2682
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/compiler/loops.jl
View File

@ -152,7 +152,7 @@ function unseqin(v::IVertex)
end end
end end
unseqout(v::IVertex) = group(v[1], map(x->x[1], inputs(v)[2:end])...) unseqout(v::IVertex) = group(v[1], v[2][1])
unseq(graph) = unseqout(unseqin(graph)) unseq(graph) = unseqout(unseqin(graph))