Skip to content

Commit 3f623df

Browse files
committed
Merge branch 'master' of https://github.com/fchollet/keras
2 parents 69e19b1 + 0ed465a commit 3f623df

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

keras/layers/containers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ def add_input(self, name, input_shape=None,
313313
if dtype == 'float':
314314
layer.input = K.placeholder(shape=layer.input_shape, name=name)
315315
else:
316-
if len(input_shape) == 1:
316+
if (input_shape and len(input_shape) == 1) or (batch_input_shape and len(batch_input_shape) == 2):
317317
layer.input = K.placeholder(shape=layer.input_shape,
318318
dtype='int32',
319319
name=name)

0 commit comments

Comments
 (0)