← Back
Editing: _manipulation_functions.cpython-311.pyc
� ���c� � � � d dl mZ ddlmZ ddlmZ d dlmZmZm Z m Z d dl Z d d�d d�Z d d�Zdd�d d�Zd!d�Zd"d�Zdd�d#d�Zd$d �Zd d�d%d �ZdS )&� )� annotations� )�Array)� result_type)�List�Optional�Tuple�UnionN��axis�arrays�%Union[Tuple[Array, ...], List[Array]]r � Optional[int]�returnr c � � t | � }t d� | D � � � � } t j t j | ||�� � � � S )z� Array API compatible wrapper for :py:func:`np.concatenate <numpy.concatenate>`. See its docstring for more information. c 3 �$ K � | ] }|j V � � d S �N��_array��.0�as �I/usr/lib/python3/dist-packages/numpy/array_api/_manipulation_functions.py� <genexpr>zconcat.<locals>.<genexpr> �$ � � � �,�,��1�8�,�,�,�,�,�,� )r �dtype)r �tupler �_new�np� concatenate)r r r s r �concatr" sK � � �� �E� �,�,�V�,�,�,� ,� ,�F� �:�b�n�V�$�e�D�D�D� E� E�Er �x�intc �Z � t j t j | j |� � � � S )z� Array API compatible wrapper for :py:func:`np.expand_dims <numpy.expand_dims>`. See its docstring for more information. )r r r � expand_dimsr �r# r s r r& r& s" � � �:�b�n�Q�X�t�4�4� 5� 5�5r �%Optional[Union[int, Tuple[int, ...]]]c �\ � t j t j | j |�� � � � S )zx Array API compatible wrapper for :py:func:`np.flip <numpy.flip>`. See its docstring for more information. r )r r r �flipr r' s r r* r* # s% � � �:�b�g�a�h�T�2�2�2� 3� 3�3r �axes�Tuple[int, ...]c �Z � t j t j | j |� � � � S )z� Array API compatible wrapper for :py:func:`np.transpose <numpy.transpose>`. See its docstring for more information. )r r r � transposer )r# r+ s r � permute_dimsr/ . s" � � �:�b�l�1�8�T�2�2� 3� 3�3r �shapec �Z � t j t j | j |� � � � S )z~ Array API compatible wrapper for :py:func:`np.reshape <numpy.reshape>`. See its docstring for more information. )r r r �reshaper )r# r0 s r r2 r2 8 s" � � �:�b�j���5�1�1� 2� 2�2r �shift�Union[int, Tuple[int, ...]]c �^ � t j t j | j ||�� � � � S )zx Array API compatible wrapper for :py:func:`np.roll <numpy.roll>`. See its docstring for more information. r )r r r �rollr )r# r3 r s r r6 r6 A s' � � �:�b�g�a�h��D�9�9�9� :� :�:r c �\ � t j t j | j |�� � � � S )z~ Array API compatible wrapper for :py:func:`np.squeeze <numpy.squeeze>`. See its docstring for more information. r )r r r �squeezer r' s r r8 r8 P s% � � �:�b�j����5�5�5� 6� 6�6r c � � t | � t d� | D � � � � } t j t j | |�� � � � S )zz Array API compatible wrapper for :py:func:`np.stack <numpy.stack>`. See its docstring for more information. c 3 �$ K � | ] }|j V � � d S r r r s r r zstack.<locals>.<genexpr>a r r r )r r r r r �stack)r r s r r; r; Y sJ � � ���� �,�,�V�,�,�,� ,� ,�F� �:�b�h�v�D�1�1�1� 2� 2�2r )r r r r r r )r# r r r$ r r )r# r r r( r r )r+ r, r# r r r )r0 r, r# r r r )r3 r4 r# r r r( r r )r r4 r# r r r )r r r r$ r r )� __future__r � _array_objectr �_data_type_functionsr �typingr r r r �numpyr r" r& r* r/ r2 r6 r8 r; � r r �<module>rB s] �� "� "� "� "� "� "� � � � � � � -� -� -� -� -� -� /� /� /� /� /� /� /� /� /� /� /� /� � � � � PQ� F� F� F� F� F� F� 6� 6� 6� 6� HL� 4� 4� 4� 4� 4� 4�4� 4� 4� 4�3� 3� 3� 3� 37� ;� ;� ;� ;� ;� ;� 7� 7� 7� 7� LM� 3� 3� 3� 3� 3� 3� 3� 3r
Save File
Cancel