관리-도구
편집 파일: _searching_functions.cpython-311.pyc
� �܋f� � �v � d dl mZ ddlmZ ddlmZmZ d dlmZm Z d dl Z ddd�dd�Z ddd�dd�Z dd�Zdd�ZdS )� )� annotations� )�Array)� _result_type�_real_numeric_dtypes)�Optional�TupleNF��axis�keepdims�xr r � Optional[int]r �bool�returnc � � | j t vrt d� � �t j t j t j | j ||�� � � � � � S )z| Array API compatible wrapper for :py:func:`np.argmax <numpy.argmax>`. See its docstring for more information. z.Only real numeric dtypes are allowed in argmaxr ) �dtyper � TypeErrorr �_new�np�asarray�argmax�_array�r r r s �w/builddir/build/BUILD/cloudlinux-venv-1.0.6/venv/lib64/python3.11/site-packages/numpy/array_api/_searching_functions.pyr r �N � � �w�*�*�*��H�I�I�I� �:�b�j���1�8�$��!R�!R�!R�S�S� T� T�T� c � � | j t vrt d� � �t j t j t j | j ||�� � � � � � S )z| Array API compatible wrapper for :py:func:`np.argmin <numpy.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 <numpy.nonzero>`. See its docstring for more information. c 3 �> K � | ]}t j |� � V � �d S )N)r r )�.0�is r � <genexpr>znonzero.<locals>.<genexpr>' s* � � � �=�=�1���A� � �=�=�=�=�=�=r )�tupler �nonzeror )r s r r& r& ! s- � � �=�=�� �1�8�(<�(<�=�=�=� =� =�=r � condition�x1�x2c �� � t |j |j � � t j ||� � \ }}t j t j | j |j |j � � � � S )zz Array API compatible wrapper for :py:func:`np.where <numpy.where>`. See its docstring for more information. )r r r �_normalize_two_argsr r �wherer )r' r( r) s r r, r, * sS � � ���2�8�$�$�$� � &�r�2� .� .�F�B�� �:�b�h�y� /���B�I�F�F� G� G�Gr )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 �<module>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