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

probably-not-working concatenation

This commit is contained in:
Mike J Innes 2016-09-06 18:04:26 +01:00
parent 8961b4c10f
commit 781f5c0c24
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -50,6 +50,9 @@ node(::typeof(softmax), xs) =
node(s::SoftmaxOutput, xs) = mx.SoftmaxOutput(data = xs, name = s.name)
node(::typeof(cat), dim::Integer, a...) = mx.Concat(a..., dim = dim)
node(::typeof(vcat), a...) = node(cat, 1, a...)
graph(vars, ::Input, x) = x
graph(vars, c::Conv, x) =