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

Adding `nest = true` option in `Tracker.gradient` 

otherwise fails and returns an error. Note that option has to be added in both `df` and `d2f`.
This commit is contained in:
Arnaud Amzallag 2019-01-24 19:29:29 -05:00 committed by GitHub
parent ca1c73ed35
commit 3cc3c463a3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/src/models/basics.md
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)
```