Array API 标准兼容性#

NumPy 2.3.0 的主命名空间以及 numpy.fft 和 numpy.linalg 命名空间都与 Python array API 标准的 2024.12 版本兼容。

NumPy 旨在实现对未来版本标准的支持——假设这些未来版本可以在给定 NumPy 的向后兼容性政策的情况下进行升级。

对于希望编写兼容 NumPy 和其他数组库代码的下游库和最终用户，我们建议参考 array API 标准本身的文档以及 SciPy 和 scikit-learn 中面向代码和开发者的文档。

请注意，为了在较旧的 NumPy 版本（< 2.0）中使用符合标准的代码，array-api-compat 包可能很有用。要测试使用 NumPy 的代码是否仅使用了符合标准的功能而非任何 NumPy 特有的功能，可以使用 array-api-strict 包。

历史

NumPy 1.22.0 是第一个通过独立的 numpy.array_api 子模块支持 array API 标准的版本。此模块被标记为实验性（导入时会发出警告），并在 NumPy 2.0 中移除，因为主命名空间已包含完整的支持（2022.12 版本 [1]）。NEP 47 和 NEP 56 描述了在 NumPy 中实现 array API 标准的动机和范围。

入口点#

NumPy 安装了一个可用于发现目的的入口点

>>> from importlib.metadata import entry_points
>>> entry_points(group='array_api', name='numpy')
[EntryPoint(name='numpy', value='numpy', group='array_api')]

请注意，省略 name='numpy' 将返回所有安装了入口点的 Array API 标准兼容实现的入口点列表。

脚注

检查#

NumPy 实现了 array API 检查实用程序。这些函数可以通过 __array_namespace_info__() 函数访问，该函数返回一个包含检查实用程序的命名空间。

__array_namespace_info__()

获取 NumPy 的 array API 检查命名空间。
本页内容