From af65b9200c5787236da70d41c84c0b6a093ece8d Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Fri, 2 Jun 2017 15:34:30 +0100 Subject: [PATCH] mxnet repeated --- 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 e3f26494..ce42cd54 100644 --- a/src/backend/mxnet/graph.jl +++ b/src/backend/mxnet/graph.jl @@ -47,6 +47,7 @@ graph(::typeof(vcat), a...) = graph(cat, 1, a...) graph(::typeof(map), f, xss::Tuple...) = map(f, xss...) graph(::typeof(getindex), t::Tuple, n::Integer) = t[n] graph(::typeof(sum), xs::Tuple) = reduce((a, b) -> graph(broadcast, +, a, b), xs) +graph(::typeof(repeated), x, n) = ntuple(_ -> x, n) a::mx.SymbolicNode ∘ b::mx.SymbolicNode = mx.broadcast_mul(a, b)