← Back
Editing: _searching_functions.cpython-311.pyc
� ���c� � �r � d dl mZ ddlmZ ddlmZ d dlmZmZ d dl Z ddd�dd�Z ddd�dd�Z dd�Z dd�ZdS )� )� annotations� )�Array)� _result_type)�Optional�TupleNF��axis�keepdims�xr r � Optional[int]r �bool�returnc � � 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. r )r �_new�np�asarray�argmax�_array�r r r s �F/usr/lib/python3/dist-packages/numpy/array_api/_searching_functions.pyr r �1 � � �:�b�j���1�8�$��!R�!R�!R�S�S� T� T�T� c � � 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. 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 �dtyper �_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 �typingr r �numpyr r r r# r* � r r �<module>r1 s� �� "� "� "� "� "� "� � � � � � � !� !� !� !� !� !� "� "� "� "� "� "� "� "� � � � � 26�� U� U� U� U� U� U� 26�� U� U� U� U� U� U�>� >� >� >� H� H� H� H� H� Hr
Save File
Cancel