combine special cases

2017-10-31 16:37:33 +00:00 · 2017-10-31 16:37:33 +00:00 · e943a39ee7
commit e943a39ee7
parent 3d8b7250ae
1 changed files with 3 additions and 1 deletions
--- a/src/layers/recurrent.jl
+++ b/src/layers/recurrent.jl
@ -1,5 +1,7 @@
 # TODO: broadcasting cat
-combine(x, h) = vcat(x, h .* trues(1, size(x, 2)))
+combine(x::AbstractMatrix, h::AbstractVector) = vcat(x, h .* trues(1, size(x, 2)))
+combine(x::AbstractVector, h::AbstractVector) = vcat(x, h)
+combine(x::AbstractMatrix, h::AbstractMatrix) = vcat(x, h)

 # Stateful recurrence