o e @sddlmZddlmZddlmZddlmZmZm Z m Z ddl Z ddd)ddZ d*ddZddd+ddZd,ddZddd-dd Zddd.d#d$Zd/d%d&Zddd0d'd(ZdS)1) 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.0arZ/var/www/html/venv/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.)copyshaper/Optional[Bool]cCsF|j}|r t|}t||}|durt||stdt|S)z~ Array API compatible wrapper for :py:func:`np.reshape `. See its docstring for more information. Fz-Incompatible shape for in-place modification.)rrr/reshape shares_memoryAttributeErrorrr)r!r0r/datareshapedrrrr28s    r2shiftUnion[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!r7r rrrr9Os r9cCr()z~ Array API compatible wrapper for :py:func:`np.squeeze `. See its docstring for more information. r )rrrsqueezerr%rrrr:^r*r:cCs.t|tdd|D}ttj||dS)zz Array API compatible wrapper for :py:func:`np.stack `. See its docstring for more information. csrrrrrrrrorzstack..r )rrrrrstack)r r rrrr;gsr;)r r r rrr)r!rr r"rr)r!rr r'rr)r+r,r!rrr)r0r,r!rr/r1rr)r7r8r!rr r'rr)r r8r!rrr)r r r r"rr) __future__r _array_objectr_data_type_functionsrtypingrrrr numpyrr r$r)r.r2r9r:r;rrrrs