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

fix matmul

This commit is contained in:
Mike J Innes 2016-09-26 19:42:24 +01:00
parent bec7219a93
commit 20c2017d15
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

@ -38,7 +38,7 @@ end
# Built-in implemenations
node(::typeof(*), args...) = mx.dot(args...)
node(::typeof(*), args...) = mx.dot(reverse(args)...)
node(::typeof(+), args...) = mx.broadcast_plus(args...)
node(::typeof(σ), x) = mx.Activation(data = x, act_type = :sigmoid)
node(::typeof(relu), x) = mx.Activation(data = x, act_type=:relu)