From 673abbcc1dad159471830f60aa78cf34a54e15c5 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Wed, 31 May 2017 18:08:11 +0100 Subject: [PATCH] mxnet hcat --- src/backend/mxnet/graph.jl | 1 + 1 file changed, 1 insertion(+) diff --git a/src/backend/mxnet/graph.jl b/src/backend/mxnet/graph.jl index 21080914..7884d4ae 100644 --- a/src/backend/mxnet/graph.jl +++ b/src/backend/mxnet/graph.jl @@ -25,6 +25,7 @@ graph(::typeof(σ), x) = mx.Activation(x, act_type = :sigmoid) graph(::typeof(relu), x) = mx.Activation(x, act_type = :relu) graph(::typeof(tanh), x) = mx.Activation(x, act_type = :tanh) graph(::typeof(flatten), x) = mx.Flatten(x) +graph(::typeof(hcat), xs...) = mx.concat(xs..., dim = 2-1) graph(::typeof(broadcast), ::typeof(+), args...) = mx.broadcast_plus(args...) graph(::typeof(broadcast), ::typeof(*), args...) = mx.broadcast_mul(args...)