TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

style

Browse Source
This commit is contained in:
Mike J Innes 2017-09-06 23:09:39 -04:00
parent cca4d25a10
commit a93c440c1e
1 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/tracker/numeric.jl
View File

@ -6,17 +6,15 @@ end
function ngradient(f, xs::AbstractArray...) function ngradient(f, xs::AbstractArray...)
grads = zeros.(xs) grads = zeros.(xs)
for (x, Δ) in zip(xs, grads) for (x, Δ) in zip(xs, grads), i in 1:length(x)
for i in 1:length(x) δ = sqrt(eps())
δ = sqrt(eps()) tmp = x[i]
tmp = x[i] x[i] = tmp - δ/2
x[i] = tmp - δ/2 y1 = f(xs...)
y1 = f(xs...) x[i] = tmp + δ/2
x[i] = tmp + δ/2 y2 = f(xs...)
y2 = f(xs...) x[i] = tmp
x[i] = tmp Δ[i] = (y2-y1)/δ
Δ[i] = (y2-y1)/δ
end
end end
return grads return grads
end end