?HAMA

Edit File Size: 3.50 KB

/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/array_api/_statistical_functions.py

Text Base64

from __future__ import annotations

from ._dtypes import (
    _real_floating_dtypes,
    _real_numeric_dtypes,
    _numeric_dtypes,
)
from ._array_object import Array
from ._dtypes import float32, float64, complex64, complex128

from typing import TYPE_CHECKING, Optional, Tuple, Union

if TYPE_CHECKING:
    from ._typing import Dtype

import numpy as np

def max(
    x: Array,
    /,
    *,
    axis: Optional[Union[int, Tuple[int, ...]]] = None,
    keepdims: bool = False,
) -> Array:
    if x.dtype not in _real_numeric_dtypes:
        raise TypeError("Only real numeric dtypes are allowed in max")
    return Array._new(np.max(x._array, axis=axis, keepdims=keepdims))

def mean(
    x: Array,
    /,
    *,
    axis: Optional[Union[int, Tuple[int, ...]]] = None,
    keepdims: bool = False,
) -> Array:
    if x.dtype not in _real_floating_dtypes:
        raise TypeError("Only real floating-point dtypes are allowed in mean")
    return Array._new(np.mean(x._array, axis=axis, keepdims=keepdims))

def min(
    x: Array,
    /,
    *,
    axis: Optional[Union[int, Tuple[int, ...]]] = None,
    keepdims: bool = False,
) -> Array:
    if x.dtype not in _real_numeric_dtypes:
        raise TypeError("Only real numeric dtypes are allowed in min")
    return Array._new(np.min(x._array, axis=axis, keepdims=keepdims))

def prod(
    x: Array,
    /,
    *,
    axis: Optional[Union[int, Tuple[int, ...]]] = None,
    dtype: Optional[Dtype] = None,
    keepdims: bool = False,
) -> Array:
    if x.dtype not in _numeric_dtypes:
        raise TypeError("Only numeric dtypes are allowed in prod")
    # Note: sum() and prod() always upcast for dtype=None. `np.prod` does that
    # for integers, but not for float32 or complex64, so we need to
    # special-case it here
    if dtype is None:
        if x.dtype == float32:
            dtype = float64
        elif x.dtype == complex64:
            dtype = complex128
    return Array._new(np.prod(x._array, dtype=dtype, axis=axis, keepdims=keepdims))

def std(
    x: Array,
    /,
    *,
    axis: Optional[Union[int, Tuple[int, ...]]] = None,
    correction: Union[int, float] = 0.0,
    keepdims: bool = False,
) -> Array:
    # Note: the keyword argument correction is different here
    if x.dtype not in _real_floating_dtypes:
        raise TypeError("Only real floating-point dtypes are allowed in std")
    return Array._new(np.std(x._array, axis=axis, ddof=correction, keepdims=keepdims))

def sum(
    x: Array,
    /,
    *,
    axis: Optional[Union[int, Tuple[int, ...]]] = None,
    dtype: Optional[Dtype] = None,
    keepdims: bool = False,
) -> Array:
    if x.dtype not in _numeric_dtypes:
        raise TypeError("Only numeric dtypes are allowed in sum")
    # Note: sum() and prod() always upcast for dtype=None. `np.sum` does that
    # for integers, but not for float32 or complex64, so we need to
    # special-case it here
    if dtype is None:
        if x.dtype == float32:
            dtype = float64
        elif x.dtype == complex64:
            dtype = complex128
    return Array._new(np.sum(x._array, axis=axis, dtype=dtype, keepdims=keepdims))

def var(
    x: Array,
    /,
    *,
    axis: Optional[Union[int, Tuple[int, ...]]] = None,
    correction: Union[int, float] = 0.0,
    keepdims: bool = False,
) -> Array:
    # Note: the keyword argument correction is different here
    if x.dtype not in _real_floating_dtypes:
        raise TypeError("Only real floating-point dtypes are allowed in var")
    return Array._new(np.var(x._array, axis=axis, ddof=correction, keepdims=keepdims))

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
tests DIR	-	drwxr-xr-x	2026-02-06 08:01:08	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2026-02-06 08:01:08	Edit Download recursive recursive
linalg.py	17.79 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
setup.py	341 B	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_array_object.py	42.71 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_constants.py	66 B	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_creation_functions.py	9.81 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_data_type_functions.py	6.14 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_dtypes.py	4.71 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_elementwise_functions.py	25.38 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_indexing_functions.py	601 B	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_manipulation_functions.py	3.24 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_searching_functions.py	1.67 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_set_functions.py	2.88 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_sorting_functions.py	1.98 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_statistical_functions.py	3.50 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_typing.py	1.20 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
_utility_functions.py	824 B	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
__init__.py	10.11 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive

Directory Contents