o ʩZh @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.0ar^/var/www/html/lang_env/lib/python3.10/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,)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-reshapeZ shares_memoryAttributeErrorrr)r r.r-dataZreshapedrrrr08s    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 r3r rrrr5Os r5cCr')z~ Array API compatible wrapper for :py:func:`np.squeeze `. See its docstring for more information. r )rrrsqueezerr$rrrr6^r)r6cCs.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 rrrr7gsr7)r r r rrr)r rr r!rr)r rr r&rr)r*r+r rrr)r.r+r rr-r/rr)r3r4r rr r&rr)r r4r rrr)r r r r!rr) __future__rZ _array_objectrZ_data_type_functionsrtypingrrrr numpyrrr#r(r,r0r5r6r7rrrrs