You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
861 B
38 lines
861 B
6 months ago
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from ._array_object import Array
|
||
|
|
|
||
|
|
from typing import Optional, Tuple, Union
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
|
||
|
|
def all(
|
||
|
|
x: Array,
|
||
|
|
/,
|
||
|
|
*,
|
||
|
|
axis: Optional[Union[int, Tuple[int, ...]]] = None,
|
||
|
|
keepdims: bool = False,
|
||
|
|
) -> Array:
|
||
|
|
"""
|
||
|
|
Array API compatible wrapper for :py:func:`np.all <numpy.all>`.
|
||
|
|
|
||
|
|
See its docstring for more information.
|
||
|
|
"""
|
||
|
|
return Array._new(np.asarray(np.all(x._array, axis=axis, keepdims=keepdims)))
|
||
|
|
|
||
|
|
|
||
|
|
def any(
|
||
|
|
x: Array,
|
||
|
|
/,
|
||
|
|
*,
|
||
|
|
axis: Optional[Union[int, Tuple[int, ...]]] = None,
|
||
|
|
keepdims: bool = False,
|
||
|
|
) -> Array:
|
||
|
|
"""
|
||
|
|
Array API compatible wrapper for :py:func:`np.any <numpy.any>`.
|
||
|
|
|
||
|
|
See its docstring for more information.
|
||
|
|
"""
|
||
|
|
return Array._new(np.asarray(np.any(x._array, axis=axis, keepdims=keepdims)))
|