Add test for reshaping unknown shapes.

keras-team · Sep 25, 2017 · 0c85f58 · 0c85f58
1 parent d1a5a66
commit 0c85f58
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():