a d @sddlmZddlmZddlmZddlmZmZm Z m Z ddl Z dddd d d d d Z d dd dddZddd dd dddZdd d dddZdd d dddZdddd dd dddZdd 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.0arp/var/www/html/stable-diffusion-webui/venv/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#)shaperr cCstt|j|S)z~ Array API compatible wrapper for :py:func:`np.reshape `. See its docstring for more information. )rrrreshaper)rr$rrrr%7sr%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'@s 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(Osr(cCs.t|tdd|D}ttj||dS)zz Array API compatible wrapper for :py:func:`np.stack `. See its docstring for more information. css|] }|jVqdSrrrrrrr`rzstack..r )rrrrrstack)r r rrrr)Xsr)) __future__rZ _array_objectrZ_data_type_functionsrtypingrrrr numpyrrrr!r#r%r'r(r)rrrrs