From fa962e61df7c7a0f177901c71bcfef4dc69e87d1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=80=9CXYangXRay=E2=80=9D?=
 <“xiaogang.yang@outlook.com”>
Date: Fri, 26 Sep 2025 10:49:23 -0400
Subject: [PATCH] fix the error of make_generator_3d

---
 ganrectf/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ganrectf/models.py b/ganrectf/models.py
index eed0a92..4127b3d 100644
--- a/ganrectf/models.py
+++ b/ganrectf/models.py
@@ -156,7 +156,7 @@ def make_generator_3d(shape_input, conv_num, conv_size, dropout, output_num):
     for fc in fc_stack:
         x = fc(x)
 
-    x = tf.reshape(x, shape=[-1, img_w, img_w, 1])
+    x = tf.reshape(x, shape=[-1, shape_input[1], shape_input[1], 1])
     # Convolutions
     for conv in conv_stack:
         x = conv(x)