diff --git a/src/layers/recurrent.jl b/src/layers/recurrent.jl index f2344af8..499a21ab 100644 --- a/src/layers/recurrent.jl +++ b/src/layers/recurrent.jl @@ -1,5 +1,5 @@ gate(h, n) = (1:h) .+ h*(n-1) -gate(x::AbstractVector, h, n) = x[gate(h,n)] +gate(x::AbstractVector, h, n) = @view x[gate(h,n)] gate(x::AbstractMatrix, h, n) = x[gate(h,n),:] # Stateful recurrence