o 0cB'@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)dtypedrY/var/www/html/gps/gps/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 rrnp _CopyMode IF_NEEDEDNotImplementedError isinstancerALWAYS_newarray_arrayint OverflowErrorasarray)rrrrr resrrrr0s  " r0)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")r2r4r)r$r rrr+r%arange)r6r2r4rrr rrrr8K r8shapeUnion[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%emptyr:rrr 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 rrrrBt rB)krrn_rowsr.n_cols Optional[int]rEcCr7)zv Array API compatible wrapper for :py:func:`np.eye `. See its docstring for more information. rrr r")MrEr)r$r rrr+r%eye)rFrGrErrr rrrrJr9rJobjectcCsddlm}|t|S)Nrr)r$r r+r% from_dlpack)r@r rrrrLs rL 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+)r:rMrrr r1rrrrOs   rOcCsXddlm}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-rrrPr+)r@rMrrr r1rrrrQs   rQT)rrendpointnumrRboolc CsDddlm}t||dvrtd||tj|||||dS)z Array API compatible wrapper for :py:func:`np.linspace `. See its docstring for more information. rrr r")rrR)r$r rrr+r%linspace)r6r2rSrrrRr rrrrUs rUxy)indexingarraysrWstr 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+)r\r,rrr szmeshgrid..cSsg|]}|jqSr)r-r[rrrr`r_rW)r$r lenrr%meshgrid)rWrXrrrrbs  rbcCr<)zx Array API compatible wrapper for :py:func:`np.ones `. See its docstring for more information. rrr r"r#)r$r rrr+r%onesr>rrrrcr?rccCrA)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-rCrrrrdrDrdrEcC4ddlm}|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 trilre)r$r rNrr+r%trilr-r@rEr rrrrh#  rhcCrf)zx Array API compatible wrapper for :py:func:`np.triu `. See its docstring for more information. rrrgz)x must be at least 2-dimensional for triure)r$r rNrr+r%triur-rirrrrk1rjrkcCr<)zz Array API compatible wrapper for :py:func:`np.zeros `. See its docstring for more information. rrr r"r#)r$r rrr+r%zerosr>rrrrl?r?rlcCrA)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-rCrrrrmRrDrm) rrrrrrrrrr )Nr) r2r3r4r5r6r5rrrrrr )r:r;rrrrrr )r@r rrrrrr r) rFr.rGrHrEr.rrrrrr )r@rKrr ) r:r;rMr5rrrrrr ) rMr5r@r rrrrrr )rSr.r6r5r2r5rrrrrRrTrr )rXr rWrYrrZ)r@r rEr.rr )% __future__rtypingrrrrr_typingr r r r r collections.abcr_dtypesrnumpyr%rr0r8r=rBrJrLrOrQrUrbrcrdrhrkrlrmrrrrsj   /