[example][add_eq] add reverse computation in example

proyan · proyan · commit 9c6836a2504e · 2021-08-19T16:18:40.000+02:00
diff --git a/example/add_eq.py b/example/add_eq.py
@@ -35,9 +35,16 @@
 dy = np.zeros(m)
 
 dx[0] = 1.
-print("x")
 dy = f.Forward(1 ,dx)
 assert(dy[0] == 2.)
 assert(dy[1] == 2.)
 
-print("y")
+# reverse computation of derivative of y[0]
+w = np.zeros((m,))
+dw = np.zeros((n),)
+
+w[0] = 1.
+w[1] = 0.
+dw = f.Reverse(1, w)
+
+assert(dw[0] == 2.)
