Merge pull request #571 from arnaudmgh/patch-1

Adding `nest = true` option in `Tracker.gradient`
This commit is contained in:
Mike J Innes 2019-01-25 09:47:48 +00:00 committed by GitHub
commit a9064cad14
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -10,12 +10,12 @@ using Flux.Tracker
f(x) = 3x^2 + 2x + 1
# df/dx = 6x + 2
df(x) = Tracker.gradient(f, x)[1]
df(x) = Tracker.gradient(f, x; nest = true)[1]
df(2) # 14.0 (tracked)
# d²f/dx² = 6
d2f(x) = Tracker.gradient(df, x)[1]
d2f(x) = Tracker.gradient(df, x; nest = true)[1]
d2f(2) # 6.0 (tracked)
```