From a93c440c1e27e54eeaf2059a3fc9be553aa864f7 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Wed, 6 Sep 2017 23:09:39 -0400 Subject: [PATCH] style --- src/tracker/numeric.jl | 20 +++++++++----------- 1 file changed, 9 insertions(+), 11 deletions(-) diff --git a/src/tracker/numeric.jl b/src/tracker/numeric.jl index 65cab8b8..d03c74c4 100644 --- a/src/tracker/numeric.jl +++ b/src/tracker/numeric.jl @@ -6,17 +6,15 @@ end function ngradient(f, xs::AbstractArray...) grads = zeros.(xs) - for (x, Δ) in zip(xs, grads) - for i in 1:length(x) - δ = sqrt(eps()) - tmp = x[i] - x[i] = tmp - δ/2 - y1 = f(xs...) - x[i] = tmp + δ/2 - y2 = f(xs...) - x[i] = tmp - Δ[i] = (y2-y1)/δ - end + for (x, Δ) in zip(xs, grads), i in 1:length(x) + δ = sqrt(eps()) + tmp = x[i] + x[i] = tmp - δ/2 + y1 = f(xs...) + x[i] = tmp + δ/2 + y2 = f(xs...) + x[i] = tmp + Δ[i] = (y2-y1)/δ end return grads end