o T=i@sxddlmZddlmZddlmZddlmZmZddl Z ddddddZ ddddddZ dddZ dddZdS)) annotations)Array) _result_type)OptionalTupleNFaxiskeepdimsxrr Optional[int]r boolreturnc Ctttj|j||dS)z| Array API compatible wrapper for :py:func:`np.argmax `. See its docstring for more information. rN)r_newnpasarrayargmax_arrayr r r rjC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\numpy/array_api/_searching_functions.pyr rc Cr)z| Array API compatible wrapper for :py:func:`np.argmin `. See its docstring for more information. rN)rrrrargminrrrrrrrrTuple[Array, ...]cCstddt|jDS)z~ Array API compatible wrapper for :py:func:`np.nonzero `. See its docstring for more information. css|]}t|VqdS)N)rr).0irrr #sznonzero..N)tuplernonzeror)r rrrrsr conditionx1x2cCs8t|j|jt||\}}tt|j|j|jS)zz Array API compatible wrapper for :py:func:`np.where `. See its docstring for more information. N)rZdtyperZ_normalize_two_argsrrwherer)r r!r"rrrr#&sr#)r rr r r r rr)r rrr)r rr!rr"rrr) __future__rZ _array_objectrZ_dtypesrtypingrrZnumpyrrrrr#rrrrs