diff options
| author | Alex Rogozhnikov <arogozhnikov@users.noreply.github.com> | 2023-02-26 07:56:47 +0000 |
|---|---|---|
| committer | Alex Rogozhnikov <arogozhnikov@users.noreply.github.com> | 2023-02-26 07:56:47 +0000 |
| commit | f07d55b27671a4575e3b9b2fc7ca9ec897d4db9e (patch) | |
| tree | 70d966f39603426d0fa9bebbeccefad5b1495834 /numpy/array_api/__init__.py | |
| parent | e141d9074e95bbdaa33ce727e9c9baa33d00875a (diff) | |
| download | numpy-f07d55b27671a4575e3b9b2fc7ca9ec897d4db9e.tar.gz | |
add support for xp.take
Diffstat (limited to 'numpy/array_api/__init__.py')
| -rw-r--r-- | numpy/array_api/__init__.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/numpy/array_api/__init__.py b/numpy/array_api/__init__.py index 5e58ee0a8..e154b9952 100644 --- a/numpy/array_api/__init__.py +++ b/numpy/array_api/__init__.py @@ -333,6 +333,10 @@ __all__ += [ "trunc", ] +from ._indexing_functions import take + +__all__ += ["take"] + # linalg is an extension in the array API spec, which is a sub-namespace. Only # a subset of functions in it are imported into the top-level namespace. from . import linalg |