From ba9a5f058f1c4f2cf8c8889745a1b01162da4fc6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Franti=C5=A1ek=20P=C3=A1rtl?= Date: Sat, 9 Apr 2022 11:47:22 +0200 Subject: [PATCH] Broadcasting replaced with comprehension in the Flux.flip function. --- src/layers/recurrent.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/recurrent.jl b/src/layers/recurrent.jl index f899f32ab1..c404ee81b0 100644 --- a/src/layers/recurrent.jl +++ b/src/layers/recurrent.jl @@ -83,7 +83,7 @@ rnn.state = hidden(rnn.cell) reset!(m::Recur) = (m.state = m.cell.state0) reset!(m) = foreach(reset!, functor(m)[1]) -flip(f, xs) = reverse(f.(reverse(xs))) +flip(f, xs) = reverse([f(x) for x in reverse(xs)]) function (m::Recur)(x::AbstractArray{T, 3}) where T h = [m(x_t) for x_t in eachslice(x, dims=3)]