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

multi-tuple output

Browse Source
This commit is contained in:
Mike J Innes 2017-05-01 16:30:24 +01:00
parent f5b0f3e530
commit 2934607115
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/backend/tensorflow/model.jl
View File

@ -18,9 +18,12 @@ function makesession(model, n::Integer; session = Session(Graph()))
makesession(model, [placeholder(Float32) for _ = 1:n], session = session)
end
retuple(xs) = xs
retuple(xs::AbstractArray{<:AbstractArray}) = (retuple.(xs)...,)
function (m::Exec)(args...)
@assert length(args) == length(m.input)
run(m.session, m.output, Dict(zip(m.input, args)))
retuple(run(m.session, m.output, Dict(zip(m.input, args))))
end
mutable struct Model