Skip to content

Commit 2ab42cd

Browse files
committed
Update the doctests of Flux.reset!
1 parent 565cf24 commit 2ab42cd

File tree

1 file changed

+18
-12
lines changed

1 file changed

+18
-12
lines changed

src/layers/recurrent.jl

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -150,23 +150,29 @@ Assuming you have a `Recur` layer `rnn`, this is roughly equivalent to:
150150
rnn.state = hidden(rnn.cell)
151151
152152
# Examples
153-
```jldoctest; filter = r"[+-]?([0-9]*[.])?[0-9]+"
154-
julia> r = RNN(1 => 1);
153+
```jldoctest
154+
julia> r = Flux.RNNCell(relu, ones(1,1), zeros(1,1), ones(1,1), zeros(1,1)); # users should use the RNN wrapper struct instead
155+
156+
julia> y = Flux.Recur(r, ones(1,1));
155157
156-
julia> a = ones(Float32, 1)
157-
1-element Vector{Float32}:
158+
julia> y.state
159+
1×1 Matrix{Float64}:
158160
1.0
159161
160-
julia> r.state
161-
1×1 Matrix{Float32}:
162-
0.0
162+
julia> y(ones(1,1)) # relu(1*1 + 1)
163+
1×1 Matrix{Float64}:
164+
2.0
163165
164-
julia> r(a); r.state
165-
1×1 Matrix{Float32}:
166-
0.61431444
166+
julia> y.state
167+
1×1 Matrix{Float64}:
168+
2.0
169+
170+
julia> Flux.reset!(y)
171+
1×1 Matrix{Float64}:
172+
0.0
167173
168-
julia> Flux.reset!(r)
169-
1×1 Matrix{Float32}:
174+
julia> y.state
175+
1×1 Matrix{Float64}:
170176
0.0
171177
```
172178
"""

0 commit comments

Comments
 (0)