{h-ie ddlmZddlmZddlmZddlmZmZm Z m Z ddl Z dd ddZ dd Zdddd Zdd Zdd  dd Zdd ddZddZddddZy)) annotations)Array) result_type)ListOptionalTupleUnionNaxisct|}td|D}tjt j |||S)z Array API compatible wrapper for :py:func:`np.concatenate `. See its docstring for more information. c34K|]}|jywN_array.0as fD:\DhanshreeandTeamAI\Github\ai_code\venv\Lib\site-packages\numpy/array_api/_manipulation_functions.py zconcat..,V188V)r dtype)rtupler_newnp concatenate)arraysr rs rconcatr s:  E ,V, ,F ::bnnV$eD EEchtjtj|j|S)z Array API compatible wrapper for :py:func:`np.expand_dims `. See its docstring for more information. )rrr expand_dimsrxr s rr"r"s" ::bnnQXXt4 55r cjtjtj|j|S)zx Array API compatible wrapper for :py:func:`np.flip `. See its docstring for more information. r )rrrfliprr#s rr&r&#s" ::bggahhT2 33r chtjtj|j|S)z Array API compatible wrapper for :py:func:`np.transpose `. See its docstring for more information. )rrr transposer)r$axess r permute_dimsr*.s" ::bll188T2 33r )copyc|j}|rtj|}tj||}|dur!tj||s t 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.)rrr+reshape shares_memoryAttributeErrorrr)r$shaper+datareshapeds rr-r-8s^ 88D wwt}zz$&H u}R--dH=LMM ::h r cltjtj|j||S)zx Array API compatible wrapper for :py:func:`np.roll `. See its docstring for more information. r )rrrrollr)r$shiftr s rr4r4Os$ ::bggahhD9 ::r cjtjtj|j|S)z~ Array API compatible wrapper for :py:func:`np.squeeze `. See its docstring for more information. r )rrrsqueezerr#s rr7r7^s" ::bjj5 66r ct|td|D}tjt j ||S)zz Array API compatible wrapper for :py:func:`np.stack `. See its docstring for more information. c34K|]}|jywrrrs rrzstack..orrr )rrrrrstack)rr s rr:r:gs6 ,V, ,F ::bhhvD1 22r )r%Union[Tuple[Array, ...], List[Array]]r z Optional[int]returnr)r$rr intr<r)r$rr %Optional[Union[int, Tuple[int, ...]]]r<r)r)Tuple[int, ...]r$rr<r)r0r?r$rr+zOptional[Bool]r<r)r5Union[int, Tuple[int, ...]]r$rr r>r<r)r r@r$rr<r)rr;r r=r<r) __future__r _array_objectr_data_type_functionsrtypingrrr r numpyrrr"r&r*r-r4r7r:r rrGs" -//PQ F 1 F?L F  F6HL44$(  " !  -2  837 ; ' ;  ; 0 ;  ;7LM 3r