pre-alloc dA as zeros

XtractOpen · Feb 6, 2018 · de6680d · de6680d
1 parent 5bb807f
commit de6680d
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/activations/tanhActivation.jl b/src/activations/tanhActivation.jl
@@ -20,11 +20,9 @@ export tanhActivation
 function tanhActivation(Y::Array{T,2},doDerivative::Bool=false) where {T <: Number}
 
     A = tanh.(Y)
-    dA = similar(Y)
+    dA = zeros(A)
     if doDerivative
          dA .= one(T) .- A.^2
-    else
-        fill!(dA, zero(T))
     end
     return A, dA
 end