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

mxnet hcat

Browse Source
This commit is contained in:
Mike J Innes 2017-05-31 18:08:11 +01:00
parent 033ed13f69
commit 673abbcc1d
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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...)