Skip to content

Commit f038cff

Browse files
Merge pull request #1938 from fpartl/fix-flip-comprehension
Broadcasting replaced with comprehension in the Flux.flip function.
2 parents 38e9fa0 + ba9a5f0 commit f038cff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/layers/recurrent.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ rnn.state = hidden(rnn.cell)
8383
reset!(m::Recur) = (m.state = m.cell.state0)
8484
reset!(m) = foreach(reset!, functor(m)[1])
8585

86-
flip(f, xs) = reverse(f.(reverse(xs)))
86+
flip(f, xs) = reverse([f(x) for x in reverse(xs)])
8787

8888
function (m::Recur)(x::AbstractArray{T, 3}) where T
8989
h = [m(x_t) for x_t in eachslice(x, dims=3)]

0 commit comments

Comments
 (0)