fixes
This commit is contained in:
parent
a14cbf301d
commit
242b4d6e23
@ -30,7 +30,7 @@ graph(::typeof(softmax), xs) =
|
||||
mx.broadcast_div(exp(xs), mx.Reshape(mx.sum(exp(xs)), shape = (1,1)))
|
||||
|
||||
graph(::typeof(cat), dim::Integer, a...) = mx.Concat(a..., dim = dim)
|
||||
graph(::typeof(vcat), a...) = node(cat, 1, a...)
|
||||
graph(::typeof(vcat), a...) = graph(cat, 1, a...)
|
||||
|
||||
graph(::Input, x) = x
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user