Array API 函数

注意

cupy.array_api 将在 CuPy v14 中移除，因为其对应的 NumPy 版本 numpy.array_api 已经移除。根模块 cupy.* 现在与 Array API 规范兼容，因为它镜像了 NumPy v2 API。请使用 Array API 兼容库 来开发与各种数组库兼容的应用程序，包括 CuPy、NumPy 和 PyTorch。

本节是已实现 API 的完整列表。详细文档请参见 Array API 规范。

cupy.array_api.abs(x, /)

与 np.abs 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.acos(x, /)

与 np.arccos 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.acosh(x, /)

与 np.arccosh 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.add(x1, x2, /)

与 np.add 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.all(x, /, *, axis=None, keepdims=False)

与 np.all 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.any(x, /, *, axis=None, keepdims=False)

与 np.any 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.arange(start, /, stop=None, step=1, *, dtype=None, device=None)

与 np.arange 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.argmax(x, /, *, axis=None, keepdims=False)

与 np.argmax 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.argmin(x, /, *, axis=None, keepdims=False)

与 np.argmin 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.argsort(x, /, *, axis=-1, descending=False, stable=True)

与 np.argsort 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.asarray(obj, /, *, dtype=None, device=None, copy=None)

与 np.asarray 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.asin(x, /)

与 np.arcsin 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.asinh(x, /)

与 np.arcsinh 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.atan(x, /)

与 np.arctan 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.atan2(x1, x2, /)

与 np.arctan2 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.atanh(x, /)

与 np.arctanh 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.bitwise_and(x1, x2, /)

与 np.bitwise_and 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.bitwise_invert(x, /)

与 np.invert 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.bitwise_left_shift(x1, x2, /)

与 np.left_shift 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.bitwise_or(x1, x2, /)

与 np.bitwise_or 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.bitwise_right_shift(x1, x2, /)

与 np.right_shift 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.bitwise_xor(x1, x2, /)

与 np.bitwise_xor 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.broadcast_arrays(*arrays)

与 np.broadcast_arrays 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

List[Array]

cupy.array_api.broadcast_to(x, /, shape)

与 np.broadcast_to 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.can_cast(from_, to, /)

与 np.can_cast 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

bool

cupy.array_api.ceil(x, /)

与 np.ceil 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.concat(arrays, /, *, axis=0)

与 np.concatenate 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.cos(x, /)

与 np.cos 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.cosh(x, /)

与 np.cosh 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.divide(x1, x2, /)

与 np.divide 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.empty(shape, *, dtype=None, device=None)

与 np.empty 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.empty_like(x, /, *, dtype=None, device=None)

与 np.empty_like 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.equal(x1, x2, /)

与 np.equal 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.exp(x, /)

与 np.exp 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.expand_dims(x, /, *, axis)

与 np.expand_dims 兼容 Array API 的包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.expm1(x, /)

与 Array API 兼容的 np.expm1 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.eye(n_rows, n_cols=None, /, *, k=0, dtype=None, device=None)

与 Array API 兼容的 np.eye 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.finfo(type, /)

与 Array API 兼容的 np.finfo 包装器。

更多信息请参见其 docstring。

返回类型:

finfo_object

cupy.array_api.flip(x, /, *, axis=None)

与 Array API 兼容的 np.flip 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.floor(x, /)

与 Array API 兼容的 np.floor 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.floor_divide(x1, x2, /)

与 Array API 兼容的 np.floor_divide 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.from_dlpack(x, /)

与 Array API 兼容的 np.from_dlpack 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.full(shape, fill_value, *, dtype=None, device=None)

与 Array API 兼容的 np.full 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.full_like(x, /, fill_value, *, dtype=None, device=None)

与 Array API 兼容的 np.full_like 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.greater(x1, x2, /)

与 Array API 兼容的 np.greater 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.greater_equal(x1, x2, /)

与 Array API 兼容的 np.greater_equal 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.iinfo(type, /)

与 Array API 兼容的 np.iinfo 包装器。

更多信息请参见其 docstring。

返回类型:

iinfo_object

cupy.array_api.isfinite(x, /)

与 Array API 兼容的 np.isfinite 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.isinf(x, /)

与 Array API 兼容的 np.isinf 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.isnan(x, /)

与 Array API 兼容的 np.isnan 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.less(x1, x2, /)

与 Array API 兼容的 np.less 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.less_equal(x1, x2, /)

与 Array API 兼容的 np.less_equal 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.linspace(start, stop, /, num, *, dtype=None, device=None, endpoint=True)

与 Array API 兼容的 np.linspace 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.log(x, /)

与 Array API 兼容的 np.log 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.log10(x, /)

与 Array API 兼容的 np.log10 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.log1p(x, /)

与 Array API 兼容的 np.log1p 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.log2(x, /)

与 Array API 兼容的 np.log2 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.logaddexp(x1, x2)

与 Array API 兼容的 np.logaddexp 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.logical_and(x1, x2, /)

与 Array API 兼容的 np.logical_and 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.logical_not(x, /)

与 Array API 兼容的 np.logical_not 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.logical_or(x1, x2, /)

与 Array API 兼容的 np.logical_or 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.logical_xor(x1, x2, /)

与 Array API 兼容的 np.logical_xor 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.matmul(x1, x2, /)

与 Array API 兼容的 np.matmul 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.meshgrid(*arrays, indexing='xy')

与 Array API 兼容的 np.meshgrid 包装器。

更多信息请参见其 docstring。

返回类型:

List[Array]

cupy.array_api.multiply(x1, x2, /)

与 Array API 兼容的 np.multiply 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.negative(x, /)

与 Array API 兼容的 np.negative 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.nonzero(x, /)

与 Array API 兼容的 np.nonzero 包装器。

更多信息请参见其 docstring。

返回类型:

Tuple[Array, …]

cupy.array_api.not_equal(x1, x2, /)

与 Array API 兼容的 np.not_equal 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.ones(shape, *, dtype=None, device=None)

与 Array API 兼容的 np.ones 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.ones_like(x, /, *, dtype=None, device=None)

与 Array API 兼容的 np.ones_like 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.permute_dims(x, /, axes)

与 Array API 兼容的 np.transpose 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.positive(x, /)

与 Array API 兼容的 np.positive 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.pow(x1, x2, /)

与 Array API 兼容的 np.power 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.remainder(x1, x2, /)

与 Array API 兼容的 np.remainder 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.reshape(x, /, shape)

与 Array API 兼容的 np.reshape 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.result_type(*arrays_and_dtypes)

与 Array API 兼容的 np.result_type 包装器。

更多信息请参见其 docstring。

返回类型:

Dtype

cupy.array_api.roll(x, /, shift, *, axis=None)

与 Array API 兼容的 np.roll 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.round(x, /)

与 Array API 兼容的 np.round 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.sign(x, /)

与 Array API 兼容的 np.sign 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.sin(x, /)

与 Array API 兼容的 np.sin 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.sinh(x, /)

与 Array API 兼容的 np.sinh 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.sort(x, /, *, axis=-1, descending=False, stable=True)

与 Array API 兼容的 np.sort 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.sqrt(x, /)

与 Array API 兼容的 np.sqrt 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.square(x, /)

与 Array API 兼容的 np.square 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.squeeze(x, /, axis)

与 Array API 兼容的 np.squeeze 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.stack(arrays, /, *, axis=0)

与 Array API 兼容的 np.stack 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.subtract(x1, x2, /)

与 Array API 兼容的 np.subtract 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.take(x, indices, /, *, axis)

与 Array API 兼容的 np.take 包装器。更多信息请参见其文档字符串。

返回类型:

Array

cupy.array_api.tan(x, /)

与 Array API 兼容的 np.tan 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.tanh(x, /)

与 Array API 兼容的 np.tanh 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.tril(x, /, *, k=0)

与 Array API 兼容的 np.tril 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.triu(x, /, *, k=0)

与 Array API 兼容的 np.triu 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.trunc(x, /)

与 Array API 兼容的 np.trunc 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.unique_all(x, /)

与 Array API 兼容的 np.unique 包装器。

更多信息请参见其 docstring。

返回类型:

UniqueAllResult

cupy.array_api.unique_inverse(x, /)

与 Array API 兼容的 np.unique 包装器。

更多信息请参见其 docstring。

返回类型:

UniqueInverseResult

cupy.array_api.unique_values(x, /)

与 Array API 兼容的 np.unique 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.where(condition, x1, x2, /)

与 Array API 兼容的 np.where 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.zeros(shape, *, dtype=None, device=None)

与 Array API 兼容的 np.zeros 包装器。

更多信息请参见其 docstring。

返回类型:

Array

cupy.array_api.zeros_like(x, /, *, dtype=None, device=None)

与 Array API 兼容的 np.zeros_like 包装器。

更多信息请参见其 docstring。

返回类型:

Array