Update GradientTape.py

laxxli · Mar 8, 2019 · 98e1b5c · 98e1b5c
1 parent ac040f2
commit 98e1b5c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/1-Introduction/GradientTape.py b/1-Introduction/GradientTape.py
@@ -21,7 +21,7 @@
 
 x = tf.constant(3.0)
 with tf.GradientTape(persistent=True) as t:
-    t.watch(x)
+    t.watch(x)       # Ensures that `tensor` is being traced by this tape.
     y = x * x
     z = y * y
 dz_dx = t.gradient(z, x)  # 108.0 (4*x^3 at x = 3)