U
|
O ±dà ã @ s d dl mZ ddlmZ ddlmZ d dlmZmZ d dl 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)ÚOptionalÚTupleNF©ÚaxisÚkeepdimsr z Optional[int]Úbool)Úxr r
|
Úreturnc C s t t tj| j||d¡¡S )z|
|
Array API compatible wrapper for :py:func:`np.argmax <numpy.argmax>`.
|
|
See its docstring for more information.
|
r )r Ú_newÚnpÚasarrayÚargmaxÚ_array©r r r
|
© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/array_api/_searching_functions.pyr s r c C s t t tj| j||d¡¡S )z|
|
Array API compatible wrapper for :py:func:`np.argmin <numpy.argmin>`.
|
|
See its docstring for more information.
|
r )r r r r Úargminr r r r r r s r zTuple[Array, ...])r r c C s t dd t | j¡D S )z~
|
Array API compatible wrapper for :py:func:`np.nonzero <numpy.nonzero>`.
|
|
See its docstring for more information.
|
c s s | ]}t |¡V qd S )N)r r )Ú.0Úir r r Ú <genexpr># s znonzero.<locals>.<genexpr>)Útupler Únonzeror )r r r r r s r )Ú conditionÚx1Úx2r c C s8 t |j|j t ||¡\}}t t | j|j|j¡¡S )zz
|
Array API compatible wrapper for :py:func:`np.where <numpy.where>`.
|
|
See its docstring for more information.
|
)r Zdtyper Z_normalize_two_argsr r Úwherer )r r r r r r r & s r )Ú
|
__future__r Z _array_objectr Z_dtypesr Útypingr r Únumpyr r r r r r r r r Ú<module> s
|
