U bB' @s:ddlmZddlmZmZmZmZmZerTddlm Z m Z m Z m Z m Z ddlmZddlmZddlZdd Zdddd d d d dddddZdKddddddd d ddddZddddd d ddddZddddd d ddddZdLdddd d!d"d!d d dd#d$d%Zd&dd'd(d)Zdddddd d dd*d+d,Zdddddd d dd-d.d/Zddd0d1d!ddd d d2dd3d4d5Zd6d7dd8d9d:d;d<Zddddd d ddd=d>Zddddd d ddd?d@Z ddAdd!ddBdCdDZ!ddAdd!ddBdEdFZ"ddddd d dddGdHZ#ddddd d dddIdJZ$dS)M) annotations) TYPE_CHECKINGListOptionalTupleUnion)ArrayDeviceDtypeNestedSequenceSupportsBufferProtocol)Sequence) _all_dtypesNcCs(dtD]}||krdSqtddS)N)Nz)dtype must be one of the supported dtypes)r ValueError)dtypedrG/tmp/pip-unpacked-wheel-cxuynqjx/numpy/array_api/_creation_functions.py_check_valid_dtypes r)rdevicecopyzZUnion[Array, bool, int, float, NestedSequence[bool | int | float], SupportsBufferProtocol]zOptional[Dtype]zOptional[Device]z#Optional[Union[bool, np._CopyMode]]r )objrrrreturncCsddlm}t||dkr*td||dtjjfkrBtdt||r|dk rb|j |krbd}|dtjj fkr| tj |j d|d S|S|dkrt|tr|d ks|d krtd tj||d }| |S)z~ Array API compatible wrapper for :py:func:`np.asarray `. See its docstring for more information. rr cpuNUnsupported device Fz!copy=False is not yet implementedNT)rrllz&Integer out of bounds for array dtypesr) _array_objectr rrnpZ _CopyModeZ IF_NEEDEDNotImplementedError isinstancerZALWAYS_newarray_arrayint OverflowErrorasarray)rrrrr resrrrr(s   "r()rrzOptional[Union[int, float]]zUnion[int, float])stopstepstartrrrcCsBddlm}t||dkr*td||tj||||dS)z| Array API compatible wrapper for :py:func:`np.arange `. See its docstring for more information. rrrr)r*r+r)rr rrr#r arange)r,r*r+rrr rrrr-Ks  r-zUnion[int, Tuple[int, ...]])shaperrrcCs>ddlm}t||dkr*td||tj||dS)zz Array API compatible wrapper for :py:func:`np.empty `. See its docstring for more information. rrrrr)rr rrr#r emptyr.rrr rrrr/as r/)xrrrcCs@ddlm}t||dkr*td||tj|j|dS)z Array API compatible wrapper for :py:func:`np.empty_like `. See its docstring for more information. rrrrr)rr rrr#r empty_liker%r1rrr rrrr2ts  r2)krrr&z Optional[int])n_rowsn_colsr4rrrcCsBddlm}t||dkr*td||tj||||dS)zv Array API compatible wrapper for :py:func:`np.eye `. See its docstring for more information. rrrr)Mr4r)rr rrr#r eye)r5r6r4rrr rrrr8s  r8object)r1rcCsddlm}|t|S)Nrr)rr r#r from_dlpack)r1r rrrr:s r:)r. fill_valuerrrcCspddlm}t||dkr*td|t||rD|jdkrD|j}tj|||d}|j t krft d| |S)zx Array API compatible wrapper for :py:func:`np.full `. See its docstring for more information. rrrrrrzInvalid input to full) rr rrr"ndimr%r fullrr TypeErrorr#)r.r;rrr r)rrrr=s  r=)r;r1rrrcCsXddlm}t||dkr*td|tj|j||d}|jtkrNt d| |S)z Array API compatible wrapper for :py:func:`np.full_like `. See its docstring for more information. rrrrrzInvalid input to full_like) rr rrr full_liker%rrr>r#)r1r;rrr r)rrrr?s  r?T)rrendpointbool)numr,r*rrr@rc CsDddlm}t||dkr*td||tj|||||dS)z Array API compatible wrapper for :py:func:`np.linspace `. See its docstring for more information. rrrr)rr@)rr rrr#r linspace)r,r*rBrrr@r rrrrCs  rCZxy)indexingstrz List[Array])arraysrDrcsRddlmtdd|Ddkr*tdfddtjdd|Dd |iDS) z Array API compatible wrapper for :py:func:`np.meshgrid `. See its docstring for more information. rrcSsh|] }|jqSrr.0arrr szmeshgrid..z,meshgrid inputs must all have the same dtypecsg|]}|qSr)r#)rHr$rrr szmeshgrid..cSsg|] }|jqSr)r%rGrrrrKsrD)rr lenrr meshgrid)rDrFrrrrMs   rMcCs>ddlm}t||dkr*td||tj||dS)zx Array API compatible wrapper for :py:func:`np.ones `. See its docstring for more information. rrrrr)rr rrr#r onesr0rrrrNs rNcCs@ddlm}t||dkr*td||tj|j|dS)z Array API compatible wrapper for :py:func:`np.ones_like `. See its docstring for more information. rrrrr)rr rrr#r ones_liker%r3rrrrOs  rOr4)r1r4rcCs4ddlm}|jdkrtd|tj|j|dS)zx Array API compatible wrapper for :py:func:`np.tril `. See its docstring for more information. rrz)x must be at least 2-dimensional for trilrP)rr r<rr#r trilr%r1r4r rrrrR#s  rRcCs4ddlm}|jdkrtd|tj|j|dS)zx Array API compatible wrapper for :py:func:`np.triu `. See its docstring for more information. rrrQz)x must be at least 2-dimensional for triurP)rr r<rr#r triur%rSrrrrT1s  rTcCs>ddlm}t||dkr*td||tj||dS)zz Array API compatible wrapper for :py:func:`np.zeros `. See its docstring for more information. rrrrr)rr rrr#r zerosr0rrrrU?s rUcCs@ddlm}t||dkr*td||tj|j|dS)z Array API compatible wrapper for :py:func:`np.zeros_like `. See its docstring for more information. rrrrr)rr rrr#r zeros_liker%r3rrrrVRs  rV)Nr)N)% __future__rtypingrrrrr_typingr r r r r collections.abcrZ_dtypesrZnumpyr rr(r-r/r2r8r:r=r?rCrMrNrOrRrTrUrVrrrrsn   /