updated batch_jacobian example that was throwing

updated batch_jacobian example that was throwing a warning
tensorflow · May 20, 2019 · 5436bd4 · 5436bd4
1 parent 53ea5d0
commit 5436bd4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow/python/eager/backprop.py b/tensorflow/python/eager/backprop.py
@@ -1112,7 +1112,7 @@ def batch_jacobian(self,
 
     ```python
     with tf.GradientTape() as g:
-      x = tf.constant([[1, 2], [3, 4]], dtype=tf.float32)
+      x = tf.constant([[1., 2.], [3., 4.]], dtype=tf.float32)
       g.watch(x)
       y = x * x
     batch_jacobian = g.batch_jacobian(y, x)