U b @sddlmZddlmZddlmZddlmZmZm Z m Z ddl Z dddd d d d d Z d dd dddZddd dd dddZdd d dddZdd d dddZdddd dd dddZdd d d d!d"Zddddd d d#d$ZdS)%) annotations)Array) result_type)ListOptionalTupleUnionNaxisz%Union[Tuple[Array, ...], List[Array]]z Optional[int]r)arraysr returncCs0t|}tdd|D}ttj|||dS)z Array API compatible wrapper for :py:func:`np.concatenate `. See its docstring for more information. css|] }|jVqdSN_array.0arK/tmp/pip-unpacked-wheel-cxuynqjx/numpy/array_api/_manipulation_functions.py szconcat..)r dtype)rtupler_newnpZ concatenate)r r rrrrconcat s rint)xr r cCstt|j|S)z Array API compatible wrapper for :py:func:`np.expand_dims `. See its docstring for more information. )rrr expand_dimsrrr rrrrsrz%Optional[Union[int, Tuple[int, ...]]]cCsttj|j|dS)zx Array API compatible wrapper for :py:func:`np.flip `. See its docstring for more information. r )rrrfliprrrrrr #sr zTuple[int, ...])axesrr cCstt|j|S)z Array API compatible wrapper for :py:func:`np.transpose `. See its docstring for more information. )rrrZ transposer)rr!rrr permute_dims.sr")shaperr cCstt|j|S)z~ Array API compatible wrapper for :py:func:`np.reshape `. See its docstring for more information. )rrrreshaper)rr#rrrr$7sr$zUnion[int, Tuple[int, ...]])shiftrr r cCsttj|j||dS)zx Array API compatible wrapper for :py:func:`np.roll `. See its docstring for more information. r )rrrrollr)rr%r rrrr&@s r&)r rr cCsttj|j|dS)z~ Array API compatible wrapper for :py:func:`np.squeeze `. See its docstring for more information. r )rrrsqueezerrrrrr'Osr'cCs.t|tdd|D}ttj||dS)zz Array API compatible wrapper for :py:func:`np.stack `. See its docstring for more information. css|] }|jVqdSrrrrrrr`szstack..r )rrrrrstack)r r rrrr(Xsr() __future__rZ _array_objectrZ_data_type_functionsrtypingrrrr Znumpyrrrr r"r$r&r'r(rrrrs