TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

overload correct repeated

Browse Source
This commit is contained in:
Mike J Innes 2017-06-19 16:48:52 -07:00
parent 7d2a34b55d
commit 5ee6e8f504
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/backend/mxnet/graph.jl
View File

@ -47,7 +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)
graph(::typeof(Base.Iterators.repeated), x, n) = ntuple(_ -> x, n)
a::mx.SymbolicNode ∘ b::mx.SymbolicNode = mx.broadcast_mul(a, b)