a hEbC' @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]}||urdSqtddS)N)Nz)dtype must be one of the supported dtypes)r ValueError)dtypedrc/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/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||dvr*td||dtjjfvrBtdt||r|durb|j |krbd}|dtjj fvr| tj |j d|d S|S|durt|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||dvr*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||dvr*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||dvr*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||dvr*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 Z _from_dlpack)r1r rrr from_dlpacks r:)r. fill_valuerrrcCspddlm}t||dvr*td|t||rD|jdkrD|j}tj|||d}|j t vrft 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||dvr*td|tj|j||d}|jtvrNt 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||dvr*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 zmeshgrid..z,meshgrid inputs must all have the same dtypecsg|]}|qSr)r#)rHr$rrr szmeshgrid..cSsg|] }|jqSr)r%rGrrrrLrKrD)rr lenrr meshgrid)rDrFrrrrNs   rNcCs>ddlm}t||dvr*td||tj||dS)zx Array API compatible wrapper for :py:func:`np.ones `. See its docstring for more information. rrrrr)rr rrr#r onesr0rrrrOs rOcCs@ddlm}t||dvr*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%r3rrrrPs  rPr4)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 trilrQ)rr r<rr#r trilr%r1r4r rrrrS#s  rScCs4ddlm}|jdkrtd|tj|j|dS)zx Array API compatible wrapper for :py:func:`np.triu `. See its docstring for more information. rrrRz)x must be at least 2-dimensional for triurQ)rr r<rr#r triur%rTrrrrU1s  rUcCs>ddlm}t||dvr*td||tj||dS)zz Array API compatible wrapper for :py:func:`np.zeros `. See its docstring for more information. rrrrr)rr rrr#r zerosr0rrrrV?s rVcCs@ddlm}t||dvr*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%r3rrrrWRs  rW)Nr)N)% __future__rtypingrrrrrZ_typingr r r r r collections.abcrZ_dtypesrnumpyr rr(r-r/r2r8r:r=r?rCrNrOrPrSrUrVrWrrrrsh   /