added tensordot (#19566)

hazemessamm · web-flow · commit 587b4c9a3c8e · 2024-04-19T21:45:06.000-07:00
diff --git a/keras/src/backend/mlx/numpy.py b/keras/src/backend/mlx/numpy.py
@@ -809,7 +809,9 @@ def tanh(x):
 
 
 def tensordot(x1, x2, axes=2):
-    raise NotImplementedError("The MLX backend doesn't support tensordot yet")
+    x1 = convert_to_tensor(x1)
+    x2 = convert_to_tensor(x2)
+    return mx.tensordot(x1, x2, axes=axes)
 
 
 def round(x, decimals=0):
