1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
U
O±d" ã@sòddlmZddlmZddlmZddlmZmZm    Z    m
Z
ddl Z ddœdd    d
d œd d „Z d
dd
dœdd„Zddœd
dd
dœdd„Zdd
d
dœdd„Zdd
d
dœdd„Zddœdd
dd
dœdd„Zdd
d
d œd!d"„Zddœddd
d œd#d$„ZdS)%é)Ú annotationsé)ÚArray)Ú result_type)ÚListÚOptionalÚTupleÚUnionN©Úaxisz%Union[Tuple[Array, ...], List[Array]]z Optional[int]r)Úarraysr ÚreturncCs0t|Ž}tdd„|Dƒƒ}t tj|||d¡S)z†
    Array API compatible wrapper for :py:func:`np.concatenate <numpy.concatenate>`.
 
    See its docstring for more information.
    css|] }|jVqdS©N©Ú_array©Ú.0Úa©rú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/array_api/_manipulation_functions.pyÚ    <genexpr>szconcat.<locals>.<genexpr>)r Údtype)rÚtuplerÚ_newÚnpZ concatenate)r r rrrrÚconcat s
rÚint)Úxr r cCst t |j|¡¡S)z†
    Array API compatible wrapper for :py:func:`np.expand_dims <numpy.expand_dims>`.
 
    See its docstring for more information.
    )rrrÚ expand_dimsr©rr rrrrsrz%Optional[Union[int, Tuple[int, ...]]]cCst tj|j|d¡S)zx
    Array API compatible wrapper for :py:func:`np.flip <numpy.flip>`.
 
    See its docstring for more information.
    r
)rrrÚfliprrrrrr #sr zTuple[int, ...])Úaxesrr cCst t |j|¡¡S)z‚
    Array API compatible wrapper for :py:func:`np.transpose <numpy.transpose>`.
 
    See its docstring for more information.
    )rrrZ    transposer)rr!rrrÚ permute_dims.sr")Úshaperr cCst t |j|¡¡S)z~
    Array API compatible wrapper for :py:func:`np.reshape <numpy.reshape>`.
 
    See its docstring for more information.
    )rrrÚreshaper)rr#rrrr$8sr$zUnion[int, Tuple[int, ...]])Úshiftrr r cCst tj|j||d¡S)zx
    Array API compatible wrapper for :py:func:`np.roll <numpy.roll>`.
 
    See its docstring for more information.
    r
)rrrÚrollr)rr%r rrrr&As r&)r rr cCst tj|j|d¡S)z~
    Array API compatible wrapper for :py:func:`np.squeeze <numpy.squeeze>`.
 
    See its docstring for more information.
    r
)rrrÚsqueezerrrrrr'Psr'cCs.t|Žtdd„|Dƒƒ}t tj||d¡S)zz
    Array API compatible wrapper for :py:func:`np.stack <numpy.stack>`.
 
    See its docstring for more information.
    css|] }|jVqdSrrrrrrraszstack.<locals>.<genexpr>r
)rrrrrÚstack)r r rrrr(Ysr()Ú
__future__rZ _array_objectrZ_data_type_functionsrÚtypingrrrr    Únumpyrrrr r"r$r&r'r(rrrrÚ<module>s   ÿ     
û