a hEb@sddlmZddlmZddlmZddlmZmZddl Z dddd d d d d d dZ dddd d d d d ddZ d ddddZ d d d d dddZdS)) annotations)Array) _result_type)OptionalTupleNFaxiskeepdimsrz Optional[int]bool)xr r returnc Cstttj|j||dS)z| Array API compatible wrapper for :py:func:`np.argmax `. See its docstring for more information. r)r_newnpasarrayargmax_arrayr r r rd/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/numpy/array_api/_searching_functions.pyr src Cstttj|j||dS)z| Array API compatible wrapper for :py:func:`np.argmin `. See its docstring for more information. r)rrrrargminrrrrrrsrzTuple[Array, ...])r r cCstddt|jDS)z~ Array API compatible wrapper for :py:func:`np.nonzero `. See its docstring for more information. css|]}t|VqdS)N)rr).0irrr #znonzero..)tuplernonzeror)r rrrrsr) conditionx1x2r cCs8t|j|jt||\}}tt|j|j|jS)zz Array API compatible wrapper for :py:func:`np.where `. See its docstring for more information. )rZdtyperZ_normalize_two_argsrrwherer)rrrrrrr &sr ) __future__rZ _array_objectrZ_dtypesrtypingrrnumpyrrrrr rrrrs