diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py
index 8a0c0b37d6bf..caef5c709f8e 100644
--- a/numpy/lib/function_base.py
+++ b/numpy/lib/function_base.py
@@ -585,8 +585,7 @@ class ndarray is returned.
 
     """
     a = asarray(a, dtype=dtype, order=order)
-    if (a.dtype.char in typecodes['AllFloat']) \
-           and (_nx.isnan(a).any() or _nx.isinf(a).any()):
+    if a.dtype.char in typecodes['AllFloat'] and not np.isfinite(a).all():
         raise ValueError(
                 "array must not contain infs or NaNs")
     return a