bug fix : cnn tensorflow backend error (#3558)

keras-team · Aug 24, 2016 · 519d1e7 · 519d1e7
1 parent 82afc71
commit 519d1e7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras/backend/tensorflow_backend.py b/keras/backend/tensorflow_backend.py
@@ -783,7 +783,7 @@ def batch_flatten(x):
     '''Turn a n-D tensor into a 2D tensor where
     the first dimension is conserved.
     '''
-    x = tf.reshape(x, [-1, prod(shape(x)[1:])])
+    x = tf.reshape(x, tf.pack([-1, prod(shape(x)[1:])]))
     return x