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

Merge pull request #571 from arnaudmgh/patch-1

Browse Source 
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
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 f(x) = 3x^2 + 2x + 1
# df/dx = 6x + 2 # 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) df(2) # 14.0 (tracked)
# d²f/dx² = 6 # 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) d2f(2) # 6.0 (tracked)
``` ```