typo

jumerckx · jumerckx · commit cfa320197e49 · 2019-06-24T20:57:53.000+02:00
diff --git a/src/softmax.jl b/src/softmax.jl
@@ -20,7 +20,7 @@ independent.
 softmax(xs; dims=1) = softmax!(similar(xs), xs, dims)
 
 function softmax!(out::AbstractArray{T}, xs::AbstractArray{T}, dims) where {T}
-    max = maximum(xs, dims)
+    max = maximum(xs, dims=dims)
     out .= exp.(xs.-max)
     out ./= sum(out, dims=dims)
     return out
