o T=i @sddlmZddlmZddlmZddlmZmZm Z m Z ddl Z ddd&ddZ d'ddZddd(ddZd)ddZd*ddZddd+d d!Zd,d"d#Zddd-d$d%ZdS).) annotations)Array) result_type)ListOptionalTupleUnionNaxisarrays%Union[Tuple[Array, ...], List[Array]]r Optional[int]returnrcCs0t|}tdd|D}ttj|||dS)z Array API compatible wrapper for :py:func:`np.concatenate `. See its docstring for more information. cs|]}|jVqdSN_array.0armC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\numpy/array_api/_manipulation_functions.py zconcat..)r dtypeN)rtupler_newnpZ concatenate)r r rrrrconcat s rxintcCtt|j|S)z Array API compatible wrapper for :py:func:`np.expand_dims `. See its docstring for more information. N)rrr expand_dimsrr r rrrr#r#%Optional[Union[int, Tuple[int, ...]]]cCttj|j|dS)zx Array API compatible wrapper for :py:func:`np.flip `. See its docstring for more information. r N)rrrfliprr$rrrr(#r(axesTuple[int, ...]cCr")z Array API compatible wrapper for :py:func:`np.transpose `. See its docstring for more information. N)rrrZ transposer)r r*rrr permute_dims.r%r,shapecCr")z~ Array API compatible wrapper for :py:func:`np.reshape `. See its docstring for more information. N)rrrreshaper)r r-rrrr.7r%r.shiftUnion[int, Tuple[int, ...]]cCsttj|j||dS)zx Array API compatible wrapper for :py:func:`np.roll `. See its docstring for more information. r N)rrrrollr)r r/r rrrr1@s r1cCr')z~ Array API compatible wrapper for :py:func:`np.squeeze `. See its docstring for more information. r N)rrrsqueezerr$rrrr2Or)r2cCs.t|tdd|D}ttj||dS)zz Array API compatible wrapper for :py:func:`np.stack `. See its docstring for more information. csrrrrrrrr`rzstack..r N)rrrrrstack)r r rrrr3Xsr3)r r r rrr)r rr r!rr)r rr r&rr)r*r+r rrr)r-r+r rrr)r/r0r rr r&rr)r r0r rrr)r r r r!rr) __future__rZ _array_objectrZ_data_type_functionsrtypingrrrr Znumpyrrr#r(r,r.r1r2r3rrrrs