Add an input validation case in Concatenate.

satels · Jan 9, 2018 · a4c1d44 · a4c1d44
1 parent f0872fc
commit a4c1d44
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras/layers/merge.py b/keras/layers/merge.py
@@ -330,9 +330,9 @@ def __init__(self, axis=-1, **kwargs):
 
     def build(self, input_shape):
         # Used purely for shape validation.
-        if not isinstance(input_shape, list):
+        if not isinstance(input_shape, list) or len(input_shape) < 2:
             raise ValueError('`Concatenate` layer should be called '
-                             'on a list of inputs')
+                             'on a list of of at least 2 inputs')
         if all([shape is None for shape in input_shape]):
             return
         reduced_inputs_shapes = [list(shape) for shape in input_shape]