Ng ddlmZddlmZddlmZddlmZmZm Z m Z ddl Z ddd!dZ d"dZddd#dZd$dZddd%dZddd&dZd'dZddd(d ZdS))) annotations)Array) result_type)ListOptionalTupleUnionNaxisarrays%Union[Tuple[Array, ...], List[Array]]r Optional[int]returnrct|}td|D}tjt j|||S)z Array API compatible wrapper for :py:func:`np.concatenate `. See its docstring for more information. c3$K|] }|jV dSN_array.0as c/var/www/html/ai-engine/env/lib/python3.11/site-packages/numpy/array_api/_manipulation_functions.py zconcat..$,,18,,,,,,)r dtype)rtupler_newnp concatenate)r r rs rconcatr" sK  E ,,V,,, , ,F :bnV$eDDD E EErxintcZtjtj|j|S)z Array API compatible wrapper for :py:func:`np.expand_dims `. See its docstring for more information. )rrr expand_dimsrr#r s rr&r&s" :bnQXt44 5 55r%Optional[Union[int, Tuple[int, ...]]]c\tjtj|j|S)zx Array API compatible wrapper for :py:func:`np.flip `. See its docstring for more information. r )rrr fliprr's rr*r*#s% :bgahT222 3 33raxesTuple[int, ...]cZtjtj|j|S)z Array API compatible wrapper for :py:func:`np.transpose `. See its docstring for more information. )rrr transposer)r#r+s r permute_dimsr/.s" :bl18T22 3 33r)copyshaper0Optional[Bool]c|j}|rtj|}tj||}|dur$tj||st dt j|S)z~ Array API compatible wrapper for :py:func:`np.reshape `. See its docstring for more information. Fz-Incompatible shape for in-place modification.)rr r0reshape shares_memoryAttributeErrorrr)r#r1r0datareshapeds rr4r48sj 8D wt}}z$&&H u}}R-dH==}LMMM :h  rshiftUnion[int, Tuple[int, ...]]c^tjtj|j||S)zx Array API compatible wrapper for :py:func:`np.roll `. See its docstring for more information. r )rrr rollr)r#r9r s rr<r<Os' :bgahD999 : ::rc\tjtj|j|S)z~ Array API compatible wrapper for :py:func:`np.squeeze `. See its docstring for more information. r )rrr squeezerr's rr>r>^s% :bj555 6 66rct|td|D}tjt j||S)zz Array API compatible wrapper for :py:func:`np.stack `. See its docstring for more information. c3$K|] }|jV dSrrrs rrzstack..orrr )rrrrr stack)r r s rrArAgsJ ,,V,,, , ,F :bhvD111 2 22r)r rr rrr)r#rr r$rr)r#rr r(rr)r+r,r#rrr)r1r,r#rr0r2rr)r9r:r#rr r(rr)r r:r#rrr)r rr r$rr) __future__r _array_objectr_data_type_functionsrtypingrrr r numpyr r"r&r*r/r4r<r>rArrrHsl"""""" ------////////////PQ F F F F F F6666HL4444444444$(       837 ; ; ; ; ; ;7777LM 3 3 3 3 3 3 3 3r