This commit is contained in:
Mike J Innes 2017-02-20 21:49:02 +00:00
parent a14cbf301d
commit 242b4d6e23

View File

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