closes #32

MikeInnes · MikeInnes · commit a93ee1a7505e · 2018-05-31T16:36:53.000+01:00
diff --git a/src/softmax.jl b/src/softmax.jl
@@ -1,6 +1,6 @@
 using Base.Threads
 
-function softmax!(out::AbstractVecOrMat, xs::AbstractVecOrMat)
+function softmax!(out::AbstractVecOrMat{T}, xs::AbstractVecOrMat{T}) where T<:AbstractFloat
   @threads for j = 1:size(xs, 2)
     @inbounds begin
       # out[end, :] .= maximum(xs, 1)
