Added arctan2 operation (#19759)

yrahul3910 · web-flow · commit 5d45a9cf67fd · 2024-05-24T22:23:13.000-07:00
diff --git a/keras/src/backend/mlx/numpy.py b/keras/src/backend/mlx/numpy.py
@@ -157,7 +157,9 @@ def arctan(x):
 
 
 def arctan2(x1, x2):
-    raise NotImplementedError("The MLX backend doesn't support arctan2 yet")
+    x1 = convert_to_tensor(x1)
+    x2 = convert_to_tensor(x2)
+    return mx.arctan2(x1, x2)
 
 
 def arctanh(x):
