ohB'DddlmZddlmZmZmZmZmZerddlm Z m Z m Z m Z m Z ddlmZddlmZddlZdZdddd  d!d Z d"ddd  d#d Zddd  d$d Zddd  d%dZ d&dddd d'dZd(dZddd  d)dZddd  d*dZdddd d+dZddd,dZddd  d$dZddd  d%dZ ddd-dZ!ddd-dZ"ddd  d$dZ#ddd  d%d Z$y).) annotations) TYPE_CHECKINGListOptionalTupleUnion)ArrayDeviceDtypeNestedSequenceSupportsBufferProtocol)Sequence) _all_dtypesNc@dtzD]}||usytd)NNz)dtype must be one of the supported dtypes)r ValueError)dtypeds ]/var/www/html/engine/venv/lib/python3.12/site-packages/numpy/array_api/_creation_functions.py_check_valid_dtypers0 { " A:  @ AA)rdevicecopycddlm}t||dvrtd||dtj j fvr tdt||re||j|k7rd}|dtj jfvr2|jt j|jd|S|S|%t|tr|d kDs|d kr td t j || }|j|S) z~ Array API compatible wrapper for :py:func:`np.asarray `. See its docstring for more information. r r cpuNUnsupported device Fz!copy=False is not yet implementedT)rrllz&Integer out of bounds for array dtypesr) _array_objectr rrnp _CopyMode IF_NEEDEDNotImplementedError isinstancerALWAYS_newarray_arrayint OverflowErrorasarray)objrrrr ress rr-r-s,%u ]".vj9:: r||--..!"EFF#u  e!3D D",,--. .5::bhhszzEJK K  }C-3=C*DTDEE **S &C 5::c?r)rrcddlm}t||dvrtd||jt j ||||S)z| Array API compatible wrapper for :py:func:`np.arange `. See its docstring for more information. r rrr)stopstepr)r!r rrr(r"arange)startr1r2rrr s rr3r3KsJ%u ]".vj9:: 5::biiDt5I JJrcddlm}t||dvrtd||jt j ||S)zz Array API compatible wrapper for :py:func:`np.empty `. See its docstring for more information. r rrrr )r!r rrr(r"emptyshaperrr s rr6r6aF%u ]".vj9:: 5::bhhuE2 33rcddlm}t||dvrtd||jt j |j|S)z Array API compatible wrapper for :py:func:`np.empty_like `. See its docstring for more information. r rrrr )r!r rrr(r" empty_liker*xrrr s rr;r;tJ%u ]".vj9:: 5::bmmAHHE: ;;r)krrcddlm}t||dvrtd||jt j ||||S)zv Array API compatible wrapper for :py:func:`np.eye `. See its docstring for more information. r rrr)Mr?r)r!r rrr(r"eye)n_rowsn_colsr?rrr s rrBrBsJ%u ]".vj9:: 5::bffVv%@ AArcXddlm}|jtj|S)Nr r)r!r r(r" from_dlpack)r=r s rrFrFs$ 5::bnnQ' ((rc$ddlm}t||dvrtd|t ||r|j dk(r |j }tj|||}|jtvr td|j|S)zx Array API compatible wrapper for :py:func:`np.full `. See its docstring for more information. r rrrrr zInvalid input to full) r!r rrr&ndimr*r"fullrr TypeErrorr()r8 fill_valuerrr r/s rrIrIs%u ]".vj9::*e$A)=&& ''%5 1C yy #/00 5::c?rcddlm}t||dvrtd|t j |j ||}|jtvr td|j|S)z Array API compatible wrapper for :py:func:`np.full_like `. See its docstring for more information. r rrrr zInvalid input to full_like) r!r rrr" full_liker*rrrJr()r=rKrrr r/s rrMrMsj%u ]".vj9:: ,,qxx5 9C yy #455 5::c?rT)rrendpointc ddlm}t||dvrtd||jt j |||||S)z Array API compatible wrapper for :py:func:`np.linspace `. See its docstring for more information. r rrr)rrN)r!r rrr(r"linspace)r4r1numrrrNr s rrPrPsL%u ]".vj9:: 5::bkk%s%(S TTrxy)indexingc$ddlm}t|Dchc]}|jc}dkDr t dt j |Dcgc]}|jc}d|iDcgc]}|j|c}Scc}wcc}wcc}w)z Array API compatible wrapper for :py:func:`np.meshgrid `. See its docstring for more information. r rz,meshgrid inputs must all have the same dtyperS) r!r lenrrr"meshgridr*r()rSarraysr ar)s rrVrVs %  V $AGG $%)GHH[[V"<188"<PxP    5  % #= sB B'B cddlm}t||dvrtd||jt j ||S)zx Array API compatible wrapper for :py:func:`np.ones `. See its docstring for more information. r rrrr )r!r rrr(r"onesr7s rrZrZsF%u ]".vj9:: 5::bgge51 22rcddlm}t||dvrtd||jt j |j|S)z Array API compatible wrapper for :py:func:`np.ones_like `. See its docstring for more information. r rrrr )r!r rrr(r" ones_liker*r<s rr\r\sJ%u ]".vj9:: 5::bll18859 ::rr?cddlm}|jdkr td|jt j |j|S)zx Array API compatible wrapper for :py:func:`np.tril `. See its docstring for more information. r rz)x must be at least 2-dimensional for trilr])r!r rHrr(r"trilr*r=r?r s rr`r`#> %vvzDEE 5::bggahh!, --rcddlm}|jdkr td|jt j |j|S)zx Array API compatible wrapper for :py:func:`np.triu `. See its docstring for more information. r rr_z)x must be at least 2-dimensional for triur])r!r rHrr(r"triur*ras rrdrd1rbrcddlm}t||dvrtd||jt j ||S)zz Array API compatible wrapper for :py:func:`np.zeros `. See its docstring for more information. r rrrr )r!r rrr(r"zerosr7s rrfrf?r9rcddlm}t||dvrtd||jt j |j|S)z Array API compatible wrapper for :py:func:`np.zeros_like `. See its docstring for more information. r rrrr )r!r rrr(r" zeros_liker*r<s rrhrhRr>r) r.zZUnion[Array, bool, int, float, NestedSequence[bool | int | float], SupportsBufferProtocol]rOptional[Dtype]rOptional[Device]rz#Optional[Union[bool, np._CopyMode]]returnr )Nr ) r1zOptional[Union[int, float]]r2Union[int, float]r4rlrrirrjrkr )r8Union[int, Tuple[int, ...]]rrirrjrkr )r=r rrirrjrkr r) rCr+rDz Optional[int]r?r+rrirrjrkr )r=objectrkr ) r8rmrKrlrrirrjrkr ) rKrlr=r rrirrjrkr )rQr+r4rlr1rlrrirrjrNboolrkr )rWr rSstrrkz List[Array])r=r r?r+rkr )% __future__rtypingrrrrr_typingr r r r rcollections.abcr_dtypesrnumpyr"rr-r3r6r;rBrFrIrMrPrVrZr\r`rdrfrhrrrxs;"?>) B,"#04) ) ) ) .) )^)- K "#K &K  K K  K K K2"# 4 &4 4  4  4(.2d <  <* <;K <  <$!B !#B B B  B  B B B,)"#  &!      @"# "      @"#U U U U  U UU U..2."# 3 &3 3  3  3(.2d ;  ;* ;;K ;  ; #$ .#$ .""# 4 &4 4  4  4(.2d <  <* <;K <