g ddlmZddlmZddlmZddlmZmZm Z m Z ddl Z ddddZ ddZddd dZd!dZd"dZddd#dZd$dZddd%dZdS)&) annotations)Array) result_type)ListOptionalTupleUnionNaxisarrays%Union[Tuple[Array, ...], List[Array]]r Optional[int]returnrct|}td|D}tjt j|||S)z Array API compatible wrapper for :py:func:`np.concatenate `. See its docstring for more information. c3$K|] }|jV dSN_array.0as ]/granary/edev/quantum/lib/python3.11/site-packages/numpy/array_api/_manipulation_functions.py zconcat..$,,18,,,,,,)r dtype)rtupler_newnp concatenate)r r rs rconcatr" sK  E ,,V,,, , ,F :bnV$eDDD E EErxintcZtjtj|j|S)z Array API compatible wrapper for :py:func:`np.expand_dims `. See its docstring for more information. )rrr expand_dimsrr#r s rr&r&s" :bnQXt44 5 55r%Optional[Union[int, Tuple[int, ...]]]c\tjtj|j|S)zx Array API compatible wrapper for :py:func:`np.flip `. See its docstring for more information. r )rrr fliprr's rr*r*#s% :bgahT222 3 33raxesTuple[int, ...]cZtjtj|j|S)z Array API compatible wrapper for :py:func:`np.transpose `. See its docstring for more information. )rrr transposer)r#r+s r permute_dimsr/.s" :bl18T22 3 33rshapecZtjtj|j|S)z~ Array API compatible wrapper for :py:func:`np.reshape `. See its docstring for more information. )rrr reshaper)r#r0s rr2r28s" :bj511 2 22rshiftUnion[int, Tuple[int, ...]]c^tjtj|j||S)zx Array API compatible wrapper for :py:func:`np.roll `. See its docstring for more information. r )rrr rollr)r#r3r s rr6r6As' :bgahD999 : ::rc\tjtj|j|S)z~ Array API compatible wrapper for :py:func:`np.squeeze `. See its docstring for more information. r )rrr squeezerr's rr8r8Ps% :bj555 6 66rct|td|D}tjt j||S)zz Array API compatible wrapper for :py:func:`np.stack `. See its docstring for more information. c3$K|] }|jV dSrrrs rrzstack..arrr )rrrrr stack)r r s rr;r;YsJ ,,V,,, , ,F :bhvD111 2 22r)r rr rrr)r#rr r$rr)r#rr r(rr)r+r,r#rrr)r0r,r#rrr)r3r4r#rr r(rr)r r4r#rrr)r rr r$rr) __future__r _array_objectr_data_type_functionsrtypingrrr r numpyr r"r&r*r/r2r6r8r;rrrBs]"""""" ------////////////PQ F F F F F F6666HL4444444444333337 ; ; ; ; ; ;7777LM 3 3 3 3 3 3 3 3r