FM / array

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

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

Name	Size	Mode	Actions
tests/	-	0755	rm
__pycache__/	-	0755	rm
linalg.py	18221	0644	edit dl rm
setup.py	341	0644	edit dl rm
_array_object.py	43739	0644	edit dl rm
_constants.py	66	0644	edit dl rm
_creation_functions.py	10050	0644	edit dl rm
_data_type_functions.py	6288	0644	edit dl rm
_dtypes.py	4823	0644	edit dl rm
_elementwise_functions.py	25992	0644	edit dl rm
_indexing_functions.py	601	0644	edit dl rm
_manipulation_functions.py	3317	0644	edit dl rm
_searching_functions.py	1715	0644	edit dl rm
_set_functions.py	2948	0644	edit dl rm
_sorting_functions.py	2031	0644	edit dl rm
_statistical_functions.py	3584	0644	edit dl rm
_typing.py	1228	0644	edit dl rm
_utility_functions.py	824	0644	edit dl rm
__init__.py	10355	0644	edit dl rm

Edit: /opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/array_api/_indexing_functions.py (601B)

from __future__ import annotations from ._array_object import Array from ._dtypes import _integer_dtypes import numpy as np def take(x: Array, indices: Array, /, *, axis: int) -> Array: """ Array API compatible wrapper for :py:func:`np.take `. See its docstring for more information. """ if indices.dtype not in _integer_dtypes: raise TypeError("Only integer dtypes are allowed in indexing") if indices.ndim != 1: raise ValueError("Only 1-dim indices array is supported") return Array._new(np.take(x._array, indices._array, axis=axis))