o &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.0arRC:\wamp64\www\opt\env\Lib\site-packages\numpy/array_api/_manipulation_functions.py zconcat..)r dtype)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. )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 )rrrfliprr$rrrr(#r(axesTuple[int, ...]cCr")z Array API compatible wrapper for :py:func:`np.transpose `. See its docstring for more information. )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. )rrrreshaper)r r-rrrr.8r%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 )rrrrollr)r r/r rrrr1As r1cCr')z~ Array API compatible wrapper for :py:func:`np.squeeze `. See its docstring for more information. r )rrrsqueezerr$rrrr2Pr)r2cCs.t|tdd|D}ttj||dS)zz Array API compatible wrapper for :py:func:`np.stack `. See its docstring for more information. csrrrrrrrrarzstack..r )rrrrrstack)r r rrrr3Ysr3)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 numpyrrr#r(r,r.r1r2r3rrrrs