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

update states references

Browse Source
This commit is contained in:
Mike J Innes 2016-10-30 01:58:39 +01:00
parent 4d45ee1bb9
commit a1b1d87767
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/backend/tensorflow/recurrent.jl
View File

@ -10,7 +10,7 @@ cgroup(xs...) = Flow.group(map(constant, xs)...)
function tf(model::Flux.Unrolled) function tf(model::Flux.Unrolled)
sess = Session(Graph()) sess = Session(Graph())
input = placeholder(Float32) input = placeholder(Float32)
instates = [placeholder(Float32) for _ in model.states] instates = [placeholder(Float32) for _ in model.state]
inputs = TensorFlow.unpack(input, num = model.steps, axis = 1) inputs = TensorFlow.unpack(input, num = model.steps, axis = 1)
params, (outstates, outputs) = tograph(model.graph, cgroup(instates...), cgroup(inputs...)) params, (outstates, outputs) = tograph(model.graph, cgroup(instates...), cgroup(inputs...))
output = TensorFlow.pack(outputs, axis = 1) output = TensorFlow.pack(outputs, axis = 1)
@ -19,7 +19,7 @@ function tf(model::Flux.Unrolled)
Model(model, sess, params, Model(model, sess, params,
[instates..., input], [outstates..., output], [instates..., input], [outstates..., output],
[gradients(output, input)]), [gradients(output, input)]),
batchone.(model.states)) batchone.(model.state))
end end
function batchseq(xs) function batchseq(xs)