Add initial array_api sub-namespace

This is based on the function stubs from the array API test suite, and is
currently based on the assumption that NumPy already follows the array API
standard. Now it needs to be modified to fix it in the places where NumPy
deviates (for example, different function names for inverse trigonometric
functions).

1 files changed, 29 insertions, 0 deletions

diff --git a/numpy/_array_api/statistical_functions.py b/numpy/_array_api/statistical_functions.py
new file mode 100644
index 000000000..2cc712aea
--- /dev/null
+++ b/numpy/_array_api/statistical_functions.py
@@ -0,0 +1,29 @@
+def max(x, /, *, axis=None, keepdims=False):
+    from .. import max
+    return max(x, axis=axis, keepdims=keepdims)
+
+def mean(x, /, *, axis=None, keepdims=False):
+    from .. import mean
+    return mean(x, axis=axis, keepdims=keepdims)
+
+def min(x, /, *, axis=None, keepdims=False):
+    from .. import min
+    return min(x, axis=axis, keepdims=keepdims)
+
+def prod(x, /, *, axis=None, keepdims=False):
+    from .. import prod
+    return prod(x, axis=axis, keepdims=keepdims)
+
+def std(x, /, *, axis=None, correction=0.0, keepdims=False):
+    from .. import std
+    return std(x, axis=axis, correction=correction, keepdims=keepdims)
+
+def sum(x, /, *, axis=None, keepdims=False):
+    from .. import sum
+    return sum(x, axis=axis, keepdims=keepdims)
+
+def var(x, /, *, axis=None, correction=0.0, keepdims=False):
+    from .. import var
+    return var(x, axis=axis, correction=correction, keepdims=keepdims)
+
+__all__ = ['max', 'mean', 'min', 'prod', 'std', 'sum', 'var']