mxnet repeated

2017-06-02 15:34:30 +01:00 · 2017-06-02 15:34:30 +01:00 · af65b9200c
commit af65b9200c
parent 4698cc0156
1 changed files with 1 additions and 0 deletions
--- 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)