From 3cc3c463a343b912860ddfda783e10be82093dbe Mon Sep 17 00:00:00 2001 From: Arnaud Amzallag Date: Thu, 24 Jan 2019 19:29:29 -0500 Subject: [PATCH] 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`. --- docs/src/models/basics.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/src/models/basics.md b/docs/src/models/basics.md index 606dac1c..55a72c2b 100644 --- a/docs/src/models/basics.md +++ b/docs/src/models/basics.md @@ -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) ```