Fix keras example

agmitre · Jul 19, 2018 · 166ca5d · 166ca5d
1 parent d50165d
commit 166ca5d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/samples/core/guide/autograph.ipynb b/samples/core/guide/autograph.ipynb
@@ -679,7 +679,7 @@
         "def collatz(x):\n",
         "  x=tf.reshape(x,())\n",
         "  assert x>0\n",
-        "  n = tf.convert_to_tensor(0) \n",
+        "  n = tf.convert_to_tensor((0,)) \n",
         "  while not tf.equal(x,1):\n",
         "    n+=1\n",
         "    if tf.equal(x%2, 0):\n",