o 0c @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.0ar]/var/www/html/gps/gps/lib/python3.10/site-packages/numpy/array_api/_manipulation_functions.py zconcat..)r dtype)rtupler_newnp concatenate)r r rrrrconcat s r xintcCtt|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. )rrr 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/rrrr07r&r0shiftUnion[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!r1r rrrr3@s r3cCr()z~ Array API compatible wrapper for :py:func:`np.squeeze `. See its docstring for more information. r )rrrsqueezerr%rrrr4Or*r4cCs.t|tdd|D}ttj||dS)zz Array API compatible wrapper for :py:func:`np.stack `. See its docstring for more information. csrrrrrrrr`rzstack..r )rrrrrstack)r r rrrr5Xsr5)r r r rrr)r!rr r"rr)r!rr r'rr)r+r,r!rrr)r/r,r!rrr)r1r2r!rr r'rr)r r2r!rrr)r r r r"rr) __future__r _array_objectr_data_type_functionsrtypingrrrr numpyrr r$r)r.r0r3r4r5rrrrs