mxnet vec
This commit is contained in:
parent
673abbcc1d
commit
c969d13306
@ -26,6 +26,7 @@ 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(vec), xs) = reshape(xs, shape = (-1,))
|
||||
|
||||
graph(::typeof(broadcast), ::typeof(+), args...) = mx.broadcast_plus(args...)
|
||||
graph(::typeof(broadcast), ::typeof(*), args...) = mx.broadcast_mul(args...)
|
||||
|
Loading…
Reference in New Issue
Block a user