Fix output_shape too.

rizar · Jan 7, 2016 · e21a6a9 · e21a6a9
1 parent bf2f64b
commit e21a6a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras/layers/core.py b/keras/layers/core.py
@@ -1376,7 +1376,7 @@ def output_shape(self):
         else:
             output_shape_func = marshal.loads(self._output_shape)
             output_shape_func = types.FunctionType(output_shape_func, globals())
-            shape = output_shape_func(self.previous.output_shape)
+            shape = output_shape_func(self.input_shape)
             if type(shape) not in {list, tuple}:
                 raise Exception('output_shape function must return a tuple')
             return tuple(shape)