| System: Linux server378.web-hosting.com 4.18.0-553.80.1.lve.el8.x86_64 #1 SMP Wed Oct 22 19:29:36 UTC 2025 x86_64 Current Path : /opt/cloudlinux/venv/lib64/python3.11/site-packages/numpy/array_api/ |
| Current File : //opt/cloudlinux/venv/lib64/python3.11/site-packages/numpy/array_api/_indexing_functions.py |
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 <numpy.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))