Remove use of _keras_shape in merge.py

SabrinaAL · Jan 8, 2018 · 3966daa · 3966daa
1 parent ab97958
commit 3966daa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras/layers/merge.py b/keras/layers/merge.py
@@ -245,7 +245,7 @@ def _merge_function(self, inputs):
         if len(inputs) != 2:
             raise ValueError('`Subtract` layer should be called '
                              'on exactly 2 inputs')
-        if inputs[0]._keras_shape != inputs[1]._keras_shape:
+        if K.int_shape(inputs[0]) != K.int_shape(inputs[1]):
             raise ValueError('`Subtract` layer should be called '
                              'on inputs of the same shape')
         return inputs[0] - inputs[1]