From f1da8233eebb5c4e79ffef1b4509f70f408a7247 Mon Sep 17 00:00:00 2001 From: Boyd Kane <33420535+beyarkay@users.noreply.github.com> Date: Mon, 14 Aug 2023 10:32:03 +0200 Subject: [PATCH] Set autoencoder's shape to x_test.shape[1:] Co-authored-by: Mark McDonald --- site/en/tutorials/generative/autoencoder.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/site/en/tutorials/generative/autoencoder.ipynb b/site/en/tutorials/generative/autoencoder.ipynb index 508b136307f..d81628fb401 100644 --- 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" ]