HV v d dl mZ ddlmZ ddlmZmZ d dlmZm Z d dl
Z dddddZ dddddZ
ddZddZdS ) ) annotations )Array) _result_type_real_numeric_dtypes)OptionalTupleNFaxiskeepdimsxr r
Optional[int]r boolreturnc | j t vrt d t j t j t j | j || S )z|
Array API compatible wrapper for :py:func:`np.argmax `.
See its docstring for more information.
z.Only real numeric dtypes are allowed in argmaxr
) dtyper TypeErrorr _newnpasarrayargmax_arrayr
r r s x/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/array_api/_searching_functions.pyr r N w***HIII :bj18$!R!R!RSS T TT c | j t vrt d t j t j t j | j || S )z|
Array API compatible wrapper for :py:func:`np.argmin `.
See its docstring for more information.
z.Only real numeric dtypes are allowed in argminr
) r r r r r r r argminr r s r r r r r Tuple[Array, ...]c b t d t j | j D S )z~
Array API compatible wrapper for :py:func:`np.nonzero `.
See its docstring for more information.
c 3 > K | ]}t j | V d S )N)r r ).0is r znonzero..' s* ==1A ======r )tupler nonzeror )r
s r r& r& ! s- ==
18(<(<=== = ==r conditionx1x2c t |j |j t j || \ }}t j t j | j |j |j S )zz
Array API compatible wrapper for :py:func:`np.where `.
See its docstring for more information.
)r r r _normalize_two_argsr r wherer )r' r( r) s r r, r, * sS 28$$$
&r2
.
.FB :bhy /BIFF G GGr )r
r r r r r r r )r
r r r )r' r r( r r) r r r )
__future__r
_array_objectr _dtypesr r typingr r numpyr r r r& r, r r r3 s " " " " " " 7 7 7 7 7 7 7 7 " " " " " " " " 26 U U U U U U 26 U U U U U U> > > > H H H H H Hr