Add isnan (#423)

2025-12-14 17:12:49 +08:00 · 2024-01-12 14:16:48 -05:00
parent 29081204d1
commit 1416e7b664
7 changed files with 72 additions and 0 deletions
--- a/python/src/array.cpp
+++ b/python/src/array.cpp
@@ -1132,6 +1132,12 @@ void init_array(py::module_& m) {
          py::kw_only(),
          "stream"_a = none,
          "See :func:`any`.")
+      .def(
+          "isnan",
+          &mlx::core::isnan,
+          py::kw_only(),
+          "stream"_a = none,
+          "See :func:`isnan`.")
      .def(
          "moveaxis",
          &moveaxis,