提交 3f623df0 编写于 作者: F Francois Chollet

Merge branch 'master' of https://github.com/fchollet/keras

......@@ -313,7 +313,7 @@ class Graph(Layer):
if dtype == 'float':
layer.input = K.placeholder(shape=layer.input_shape, name=name)
else:
if len(input_shape) == 1:
if (input_shape and len(input_shape) == 1) or (batch_input_shape and len(batch_input_shape) == 2):
layer.input = K.placeholder(shape=layer.input_shape,
dtype='int32',
name=name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册