`iW:ddlmZddlmZddlmZddlZd d ZdS)) annotations)Array)_integer_dtypesNxrindicesaxisintreturnc|jtvrtd|jdkrt dt jtj|j |j |S)zw Array API compatible wrapper for :py:func:`np.take `. See its docstring for more information. z+Only integer dtypes are allowed in indexingrz%Only 1-dim indices array is supported)r ) dtyper TypeErrorndim ValueErrorr_newnptake_array)rrr s v/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/array_api/_indexing_functions.pyrrsa }O++EFFF|q@AAA :bgahTBBB C CC)rrrrr r r r) __future__r _array_objectr_dtypesrcupyrrrrrsk"""""" $$$$$$ D D D D D Dr