o ʩZhB'@sddlmZddlmZmZmZmZmZer*ddlm Z m Z m Z m Z m Z ddlmZddlmZddlZdd Zdddd dRddZ  dSddddTddZddddUd!d"ZddddVd$d%Z dWdddd&dXd,d-ZdYd/d0ZddddZd2d3Zdddd[d4d5Zddd6d7d\d;d<Zd=d>d]dCdDZddddUdEdFZddddVdGdHZ ddId^dJdKZ!ddId^dLdMZ"ddddUdNdOZ#ddddVdPdQZ$dS)_) annotations) TYPE_CHECKINGListOptionalTupleUnion)ArrayDeviceDtypeNestedSequenceSupportsBufferProtocol)Sequence) _all_dtypesNcCs$dtD] }||ur dSqtd)NNz)dtype must be one of the supported dtypes)r ValueError)dtypedrZ/var/www/html/lang_env/lib/python3.10/site-packages/numpy/array_api/_creation_functions.py_check_valid_dtypes r)rdevicecopyobjZUnion[Array, bool, int, float, NestedSequence[bool | int | float], SupportsBufferProtocol]rOptional[Dtype]rOptional[Device]r#Optional[Union[bool, np._CopyMode]]returnr cCsddlm}t||dvrtd||dtjjfvr!tdt||rG|dur1|j |kr1d}|dtjj fvrE| tj |j d|d S|S|dur\t|tr\|d ksX|d kr\td 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-)rrstopOptional[Union[int, float]]stepUnion[int, float]startcCBddlm}t||dvrtd||tj||||dS)z| Array API compatible wrapper for :py:func:`np.arange `. See its docstring for more information. rrr r")r/r1r)r$r rrr(r%arange)r3r/r1rrr rrrr5K r5shapeUnion[int, Tuple[int, ...]]cC>ddlm}t||dvrtd||tj||dS)zz Array API compatible wrapper for :py:func:`np.empty `. See its docstring for more information. rrr r"r#)r$r rrr(r%emptyr7rrr rrrr:a r:xcC@ddlm}t||dvrtd||tj|j|dS)z Array API compatible wrapper for :py:func:`np.empty_like `. See its docstring for more information. rrr r"r#)r$r rrr(r% empty_liker*r=rrr rrrr?t r?)krrn_rowsr+n_cols Optional[int]rBcCr4)zv Array API compatible wrapper for :py:func:`np.eye `. See its docstring for more information. rrr r")MrBr)r$r rrr(r%eye)rCrDrBrrr rrrrGr6rGobjectcCsddlm}|t|S)Nrr)r$r r(r% from_dlpack)r=r rrrrIs rI fill_valuecCspddlm}t||dvrtd|t||r"|jdkr"|j}tj|||d}|j t vr3t d| |S)zx Array API compatible wrapper for :py:func:`np.full `. See its docstring for more information. rrr r"rr#zInvalid input to full) r$r rrr'ndimr*r%fullrr TypeErrorr()r7rJrrr r.rrrrLs   rLcCsXddlm}t||dvrtd|tj|j||d}|jtvr't d| |S)z Array API compatible wrapper for :py:func:`np.full_like `. See its docstring for more information. rrr r"r#zInvalid input to full_like) r$r rrr% full_liker*rrrMr()r=rJrrr r.rrrrNs   rNT)rrendpointnumrOboolc CsDddlm}t||dvrtd||tj|||||dS)z Array API compatible wrapper for :py:func:`np.linspace `. See its docstring for more information. rrr r")rrO)r$r rrr(r%linspace)r3r/rPrrrOr rrrrRs rRZxy)indexingarraysrSstr List[Array]csRddlmtdd|Ddkrtdfddtjdd|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()rXr)rrr szmeshgrid..cSsg|]}|jqSr)r*rWrrrr\r[rS)r$r lenrr%meshgrid)rSrTrrrr^s  r^cCr9)zx Array API compatible wrapper for :py:func:`np.ones `. See its docstring for more information. rrr r"r#)r$r rrr(r%onesr;rrrr_r<r_cCr>)z Array API compatible wrapper for :py:func:`np.ones_like `. See its docstring for more information. rrr r"r#)r$r rrr(r% ones_liker*r@rrrr`rAr`rBcC4ddlm}|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 trilra)r$r rKrr(r%trilr*r=rBr rrrrd#  rdcCrb)zx Array API compatible wrapper for :py:func:`np.triu `. See its docstring for more information. rrrcz)x must be at least 2-dimensional for triura)r$r rKrr(r%triur*rerrrrg1rfrgcCr9)zz Array API compatible wrapper for :py:func:`np.zeros `. See its docstring for more information. rrr r"r#)r$r rrr(r%zerosr;rrrrh?r<rhcCr>)z Array API compatible wrapper for :py:func:`np.zeros_like `. See its docstring for more information. rrr r"r#)r$r rrr(r% zeros_liker*r@rrrriRrAri) rrrrrrrrrr )Nr) r/r0r1r2r3r2rrrrrr )r7r8rrrrrr )r=r rrrrrr r) rCr+rDrErBr+rrrrrr )r=rHrr ) r7r8rJr2rrrrrr ) rJr2r=r rrrrrr )rPr+r3r2r/r2rrrrrOrQrr )rTr rSrUrrV)r=r rBr+rr )% __future__rtypingrrrrrZ_typingr r r r r collections.abcrZ_dtypesrnumpyr%rr-r5r:r?rGrIrLrNrRr^r_r`rdrgrhrirrrrsj   /