grddlmZddlmZddlmZddlmZmZddl Z dddddZ dddddZ ddZ ddZdS)) annotations)Array) _result_type)OptionalTupleNFaxiskeepdimsxrr Optional[int]r boolreturnc tjtjtj|j||S)z| Array API compatible wrapper for :py:func:`np.argmax `. See its docstring for more information. r )r_newnpasarrayargmax_arrayr r r s Z/granary/edev/quantum/lib/python3.11/site-packages/numpy/array_api/_searching_functions.pyrr 1 :bj18$!R!R!RSS T TTc tjtjtj|j||S)z| Array API compatible wrapper for :py:func:`np.argmin `. See its docstring for more information. r )rrrrargminrrs rrrrrTuple[Array, ...]cbtdtj|jDS)z~ Array API compatible wrapper for :py:func:`np.nonzero `. See its docstring for more information. c3>K|]}tj|VdS)N)rr).0is r znonzero..#s*==1A======r)tuplernonzeror)r s rr#r#s- == 18(<(<=== = ==r conditionx1x2ct|j|jtj||\}}tjt j|j|j|jS)zz Array API compatible wrapper for :py:func:`np.where `. See its docstring for more information. )rdtyper_normalize_two_argsrrwherer)r$r%r&s rr*r*&sS28$$$  &r2 . .FB :bhy/BIFF G GGr)r rr r r rrr)r rrr)r$rr%rr&rrr) __future__r _array_objectr_dtypesrtypingrrnumpyrrrr#r*rrr1s"""""" !!!!!!""""""""26UUUUUU26UUUUUU>>>> H H H H H Hr