Fix batch_dot of CNTK when axes=None (keras-team#9921)

Vijayabhaskar96 · May 1, 2018 · 30bf315 · 30bf315
1 parent 3b9133d
commit 30bf315
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/keras/backend/cntk_backend.py b/keras/backend/cntk_backend.py
@@ -532,7 +532,11 @@ def batch_dot(x, y, axes=None):
         axes = [len(x_shape) - 1, len(y_shape) - 2]
 
     if len(x_shape) == 2 and len(y_shape) == 2:
-        return sum(x * y, axis=1, keepdims=True)
+        if axes[0] == axes[1]:
+            result = sum(x * y, axis=axes[0], keepdims=True)
+            return result if axes[0] == 1 else transpose(result)
+        else:
+            return sum(x * transpose(y), axis=axes[0], keepdims=True)
     else:
         if len(y_shape) == 2:
             y = expand_dims(y)