Add test for reshaping unknown shapes.

keras-team · Sep 23, 2017 · 281723f · 281723f
1 parent cba3802
commit 281723f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/keras/layers/core_test.py b/tests/keras/layers/core_test.py
@@ -79,6 +79,10 @@ def test_reshape():
                kwargs={'target_shape': (1, -1)},
                input_shape=(3, 2, 4))
 
+    layer_test(layers.Reshape,
+               kwargs={'target_shape': (-1, 1)},
+               input_shape=(None, None, 4))
+
 
 @keras_test
 def test_permute():