Skip to content

Commit 1f3915d

Browse files
authored
Merge pull request #1873 from mkschleg/rnn_eachslice
Changed view to eachslice for folding in recurrent
2 parents 3cc9067 + c4338af commit 1f3915d

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
@@ -85,7 +85,7 @@ reset!(m) = foreach(reset!, functor(m)[1])
8585
flip(f, xs) = reverse(f.(reverse(xs)))
8686

8787
function (m::Recur)(x::AbstractArray{T, 3}) where T
88-
h = [m(view(x, :, :, i)) for i in 1:size(x, 3)]
88+
h = [m(x_t) for x_t in eachslice(x, dims=3)]
8989
sze = size(h[1])
9090
reshape(reduce(hcat, h), sze[1], sze[2], length(h))
9191
end

0 commit comments

Comments
 (0)