search

dmlc / keras

Deep Learning library for Python. Convnets, recurrent neural networks, and more. Runs on MXNet, Theano or TensorFlow.
http://keras.io/
Other
125 stars 34 forks source link

Fix keras shape inference for lambda outputs #84

Closed sandeep-krishnamurthy closed 6 years ago

sandeep-krishnamurthy commented 6 years ago

Fixing the issue described here - https://github.com/dmlc/keras/issues/83

After applying lambda, we need to infer the output shape from input shape, else, _keras_shape is not updated along with tensor.shape.