X7ROOT File Manager
Current Path:
/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/array_api
opt
/
cloudlinux
/
venv
/
lib
/
python3.11
/
site-packages
/
numpy
/
array_api
/
📁
..
📄
__init__.py
(10.11 KB)
📁
__pycache__
📄
_array_object.py
(42.71 KB)
📄
_constants.py
(66 B)
📄
_creation_functions.py
(9.81 KB)
📄
_data_type_functions.py
(6.14 KB)
📄
_dtypes.py
(4.71 KB)
📄
_elementwise_functions.py
(25.38 KB)
📄
_indexing_functions.py
(601 B)
📄
_manipulation_functions.py
(3.24 KB)
📄
_searching_functions.py
(1.67 KB)
📄
_set_functions.py
(2.88 KB)
📄
_sorting_functions.py
(1.98 KB)
📄
_statistical_functions.py
(3.5 KB)
📄
_typing.py
(1.2 KB)
📄
_utility_functions.py
(824 B)
📄
linalg.py
(17.79 KB)
📄
setup.py
(341 B)
📁
tests
Editing: _sorting_functions.py
from __future__ import annotations from ._array_object import Array from ._dtypes import _real_numeric_dtypes import numpy as np # Note: the descending keyword argument is new in this function def argsort( x: Array, /, *, axis: int = -1, descending: bool = False, stable: bool = True ) -> Array: """ Array API compatible wrapper for :py:func:`np.argsort <numpy.argsort>`. See its docstring for more information. """ if x.dtype not in _real_numeric_dtypes: raise TypeError("Only real numeric dtypes are allowed in argsort") # Note: this keyword argument is different, and the default is different. kind = "stable" if stable else "quicksort" if not descending: res = np.argsort(x._array, axis=axis, kind=kind) else: # As NumPy has no native descending sort, we imitate it here. Note that # simply flipping the results of np.argsort(x._array, ...) would not # respect the relative order like it would in native descending sorts. res = np.flip( np.argsort(np.flip(x._array, axis=axis), axis=axis, kind=kind), axis=axis, ) # Rely on flip()/argsort() to validate axis normalised_axis = axis if axis >= 0 else x.ndim + axis max_i = x.shape[normalised_axis] - 1 res = max_i - res return Array._new(res) # Note: the descending keyword argument is new in this function def sort( x: Array, /, *, axis: int = -1, descending: bool = False, stable: bool = True ) -> Array: """ Array API compatible wrapper for :py:func:`np.sort <numpy.sort>`. See its docstring for more information. """ if x.dtype not in _real_numeric_dtypes: raise TypeError("Only real numeric dtypes are allowed in sort") # Note: this keyword argument is different, and the default is different. kind = "stable" if stable else "quicksort" res = np.sort(x._array, axis=axis, kind=kind) if descending: res = np.flip(res, axis=axis) return Array._new(res)
Upload File
Create Folder