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

beyarkay · markmcd · web-flow · commit f1da8233eebb · 2023-08-14T10:32:03.000+02:00
Co-authored-by: Mark McDonald &lt;macd@google.com&gt;
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"
       ]

Original file line number	Diff line number	Diff line change
`@@ -179,7 +179,7 @@`
`179`	`179`	`" return decoded\n",`
`180`	`180`	`"\n",`
`181`	`181`	`"\n",`
`182`		`- "shape = x_test.shape\n",`
	`182`	`+ "shape = x_test.shape[1:]\n",`
`183`	`183`	`"latent_dim = 64\n",`
`184`	`184`	`"autoencoder = Autoencoder(latent_dim, shape)\n"`
`185`	`185`	`]`