summaryrefslogtreecommitdiff
path: root/numpy/_array_api/statistical_functions.py
blob: b9180a863f9e04cd02bb1848e25edfa35bc151f8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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)

__all__ = ['max', 'mean', 'min', 'prod', 'std', 'sum', 'var']