a ;e @sddlmZddlmZddlmZddlmZmZm Z m Z ddl Z dddd d d d d Z d dd dddZddd dd dddZdd d dddZdddd dd dddZdddd dd dd d!Zdd 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.0ar[/var/www/sunrise/env/lib/python3.9/site-packages/numpy/array_api/_manipulation_functions.py zconcat..)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 )rrrfliprr rrrr!#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#)copyzOptional[Bool])shaperr$r cCsF|j}|rt|}t||}|dur`. See its docstring for more information. Fz-Incompatible shape for in-place modification.)rrr$reshapeZ shares_memoryAttributeErrorrr)rr%r$dataZreshapedrrrr&8s   r&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*Os r*)r rr cCsttj|j|dS)z~ Array API compatible wrapper for :py:func:`np.squeeze `. See its docstring for more information. r )rrrsqueezerr rrrr+^sr+cCs.t|tdd|D}ttj||dS)zz Array API compatible wrapper for :py:func:`np.stack `. See its docstring for more information. css|] }|jVqdSrrrrrrrorzstack..r )rrrrrstack)r r rrrr,gsr,) __future__rZ _array_objectrZ_data_type_functionsrtypingrrrr numpyrrrr!r#r&r*r+r,rrrrs