Set autoencoder's shape to x_test.shape[1:]

Co-authored-by: Mark McDonald <[email protected]>
tensorflow · Aug 14, 2023 · f1da823 · f1da823
1 parent c618f46
commit f1da823
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/site/en/tutorials/generative/autoencoder.ipynb b/site/en/tutorials/generative/autoencoder.ipynb
@@ -179,7 +179,7 @@
         "    return decoded\n",
         "\n",
         "\n",
-        "shape = x_test.shape\n",
+        "shape = x_test.shape[1:]\n",
         "latent_dim = 64\n",
         "autoencoder = Autoencoder(latent_dim, shape)\n"
       ]