diff --git a/PaddleCV/gan/network/CGAN_network.py b/PaddleCV/gan/network/CGAN_network.py
index c15d104522eed90e0bc475f11e5cfc2b7cab847e..60c41b5321f9c642378b669a818f62679a43c288 100644
--- a/PaddleCV/gan/network/CGAN_network.py
+++ b/PaddleCV/gan/network/CGAN_network.py
@@ -42,7 +42,7 @@ class CGAN_model(object):
 
     def network_G(self, input, label, name="generator"):
         # concat noise and label
-        y = fluid.layers.reshape(label, shape=[-1, self.y_dim, 1, 1])
+        y = fluid.layers.reshape(label, shape=[-1, self.y_dim])
         xy = fluid.layers.concat([input, y], 1)
         o_l1 = linear(
             input=xy,