qΪ d dl mZ ddlmZ ddlmZ d dlmZmZm Z m
Z
d dl Z d dd!dZ
d"dZddd#dZd$dZddd%dZddd&d Zd'd Zd dd(d ZdS )) ) annotations )Array) result_type)ListOptionalTupleUnionNaxisarrays%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 `.
See its docstring for more information.
c 3 $ K | ] }|j V d S N_array.0as {/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/array_api/_manipulation_functions.py zconcat.. $ ,,18,,,,,, )r dtype)r tupler _newnp concatenate)r
r r s r concatr" sK
E
,,V,,,
,
,F :bnV$eDDD E EEr xintc Z t j t j | j | S )z
Array API compatible wrapper for :py:func:`np.expand_dims `.
See its docstring for more information.
)r r r expand_dimsr r# r s r r& r& s" :bnQXt44 5 55r %Optional[Union[int, Tuple[int, ...]]]c \ t j t j | j | S )zx
Array API compatible wrapper for :py:func:`np.flip `.
See its docstring for more information.
r )r r r flipr r' s r r* r* # s% :bgahT222 3 33r axesTuple[int, ...]c Z t j t j | j | S )z
Array API compatible wrapper for :py:func:`np.transpose `.
See its docstring for more information.
)r r r transposer )r# r+ s r permute_dimsr/ . s" :bl18T22 3 33r )copyshaper0 Optional[Bool]c | j }|rt j | }t j || }|du r$t j || st d t
j | S )z~
Array API compatible wrapper for :py:func:`np.reshape `.
See its docstring for more information.
Fz-Incompatible shape for in-place modification.)r r r0 reshape
shares_memoryAttributeErrorr r )r# r1 r0 datareshapeds r r4 r4 8 sj
8D wt}}z$&&H u}}R -dH = =} LMMM :h r shiftUnion[int, Tuple[int, ...]]c ^ t j t j | j || S )zx
Array API compatible wrapper for :py:func:`np.roll `.
See its docstring for more information.
r )r r r rollr )r# r9 r s r r< r< O s' :bgahD999 : ::r c \ t j t j | j | S )z~
Array API compatible wrapper for :py:func:`np.squeeze `.
See its docstring for more information.
r )r r r squeezer r' s r r> r> ^ s% :bj555 6 66r c t | t d | D } t j t j | | S )zz
Array API compatible wrapper for :py:func:`np.stack `.
See its docstring for more information.
c 3 $ K | ] }|j V d S r r r s r r zstack..o r r r )r r r r r stack)r
r s r rA rA g sJ
,,V,,,
,
,F :bhvD111 2 22r )r
r r r r r )r# r r r$ r r )r# r r r( r r )r+ r, r# r r r )r1 r, r# r r0 r2 r r )r9 r: r# r r r( r r )r r: 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/ r4 r< r> rA r r rH sl " " " " " " - - - - - - / / / / / / / / / / / / PQ F F F F F F 6 6 6 6 HL 4 4 4 4 4 44 4 4 4 $( 8 37 ; ; ; ; ; ; 7 7 7 7 LM 3 3 3 3 3 3 3 3r