Specify the behavior of clip() when one of the operands is nan

data-apis · Jun 5, 2024 · 2f6b865 · 2f6b865
1 parent 25e7177
commit 2f6b865
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/array_api_stubs/_2023_12/elementwise_functions.py b/src/array_api_stubs/_2023_12/elementwise_functions.py
@@ -807,6 +807,12 @@ def clip(
     - If a broadcasted element in ``min`` is greater than a corresponding broadcasted element in ``max``, behavior is unspecified and thus implementation-dependent.
     - If ``x`` and either ``min`` or ``max`` have different data type kinds (e.g., integer versus floating-point), behavior is unspecified and thus implementation-dependent.
 
+    **Special cases**
+
+    - If ``x_i`` is ``NaN``, the result is ``NaN``.
+    - If ``min_i`` is ``NaN``, the result is ``NaN``.
+    - If ``max_i`` is ``NaN``, the result is ``NaN``.
+
     .. versionadded:: 2023.12
     """
 

diff --git a/src/array_api_stubs/_draft/elementwise_functions.py b/src/array_api_stubs/_draft/elementwise_functions.py
@@ -808,6 +808,12 @@ def clip(
     - If a broadcasted element in ``min`` is greater than a corresponding broadcasted element in ``max``, behavior is unspecified and thus implementation-dependent.
     - If ``x`` and either ``min`` or ``max`` have different data type kinds (e.g., integer versus floating-point), behavior is unspecified and thus implementation-dependent.
 
+    **Special cases**
+
+    - If ``x_i`` is ``NaN``, the result is ``NaN``.
+    - If ``min_i`` is ``NaN``, the result is ``NaN``.
+    - If ``max_i`` is ``NaN``, the result is ``NaN``.
+
     .. versionadded:: 2023.12
     """