diff options
Diffstat (limited to 'numpy/_array_api/_statistical_functions.py')
-rw-r--r-- | numpy/_array_api/_statistical_functions.py | 29 |
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..339835095 --- /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 + # Note: the keyword argument correction is different here + return std(x, axis=axis, ddof=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 + # Note: the keyword argument correction is different here + return var(x, axis=axis, ddof=correction, keepdims=keepdims) |