note about unroll options
This commit is contained in:
parent
d73e962da9
commit
93ed44b31a
@ -85,6 +85,11 @@ function stateout(steps, offset, default)
|
|||||||
end
|
end
|
||||||
|
|
||||||
function unrollgraph(v::IVertex, n; seq = true, stateful = true)
|
function unrollgraph(v::IVertex, n; seq = true, stateful = true)
|
||||||
|
# Input: (hidden1, hidden2, ...), (x1, x2, ...)
|
||||||
|
# Output: (hidden1, hidden2, ...), (y1, y2, ...)
|
||||||
|
# If `seq` is false, takes a single `x` and uses this for each iteration.
|
||||||
|
# If `stateful` is false there are no hidden inputs or outputs.
|
||||||
|
|
||||||
state, offset, default = collect_state(v)
|
state, offset, default = collect_state(v)
|
||||||
v = group(group(state...), v)
|
v = group(group(state...), v)
|
||||||
steps = create_steps(v, n, seq = seq, stateful = stateful)
|
steps = create_steps(v, n, seq = seq, stateful = stateful)
|
||||||
|
Loading…
Reference in New Issue
Block a user