`i9ddlZddlZddlZddlmZddZ ddZ dd Z dd Zd ZGd de Z e dZ e dZ ej dddZej dddZej dddZdS)N)_corec|+td|||fDrt}nt}||}d}|d}ttj||z |z }|dkrt jd|Stj|jtj kra|dkrtd|dkr%t j |||z gtj St j |gtj St j|f|}tj|j}t|||||||S) aeReturns an array with evenly spaced values within a given interval. Values are generated within the half-open interval [start, stop). The first three arguments are mapped like the ``range`` built-in function, i.e. start and step are optional. Args: start: Start of the interval. stop: End of the interval. step: Step width between each pair of consecutive values. dtype: Data type specifier. It is inferred from other arguments by default. Returns: cupy.ndarray: The 1-D array of range values. .. seealso:: :func:`numpy.arange` Nc3jK|].}tjt|jdkV/dS)fN)numpydtypetypekind).0vals i/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_creation/ranges.py zarange..sN//{499%%*c1//////rr)rr zMarange() is only supported for booleans when the result has at most length 2.) anyfloatintrceilcupyemptyr r bool_ TypeErrorarray _arange_ufunc)startstopstepr sizerettyps raranger# ss( } // $-/// / / EEE | | uz4%<4/00 1 1D qyyz$e,,,, {5%+-- !888  199:uedl35;GGG G:ugU[999 9 *dWE * * *C +e   !C##e**cc$iiE:::: Jr2TFcJtj||t|}||}tj|f|}|r|dz n|}|dkr+|dkr||td} nXt||z |z } t|}| dkrt |||z ||nt || ||r||d<tj|tjrtj ||| |d }|r|| fS|S) aReturns an array with evenly-spaced values within a given interval. Instead of specifying the step width like :func:`cupy.arange`, this function requires the total number of elements specified. Args: start: Start of the interval. stop: End of the interval. num: Number of elements. endpoint (bool): If ``True``, the stop value is included as the last element. Otherwise, the stop value is omitted. retstep (bool): If ``True``, this function returns (array, step). Otherwise, it returns only the array. dtype: Data type specifier. It is inferred from the start and stop arguments by default. Returns: cupy.ndarray: The 1-D array of ranged values. NrrrnanoutFcopy) r result_typerrfill_linspace_ufunc_underflow_linspace_ufunc issubdtypeintegerfloorastype) rrnumendpointretstepr dtr!divrs r_linspace_scalarr:@s=,  %uSzz 2 2B } *cV2 & & &C (377SC axx 77 HHUOOOU||TE\""S(T{{ 3;; %eTE\3 D D D D E4 - - -  CG udl++! 3C    **U* ' 'CDy rcJ|dkrtd|r|dz n|}tj|}tj|} |r| rt||||||S|sAt |tjr|jjdkstj|dz}| sAt |tjr|jjdkstj|dz}tj ||t|} || }||z } tj |f| } tdd| | | d d | jzz} |dkr.| |z } tj| dkr | |z} | | z} n| | z} ntd } | | z} | |z } |r |dkr|| d <|dkrtj| d|} tj|tjrtj| | | |d} |r| | fS| S)aReturns an array with evenly-spaced values within a given interval. Instead of specifying the step width like :func:`cupy.arange`, this function requires the total number of elements specified. Args: start (scalar or array_like): Starting value(s) of the sequence. stop (scalar or array_like): Ending value(s) of the sequence, unless ``endpoint`` is set to ``False``. In that case, the sequence consists of all but the last of ``num + 1`` evenly spaced samples, so that ``stop`` is excluded. Note that the step size changes when ``endpoint`` is ``False``. num: Number of elements. endpoint (bool): If ``True``, the stop value is included as the last element. Otherwise, the stop value is omitted. retstep (bool): If ``True``, this function returns (array, step). Otherwise, it returns only the array. dtype: Data type specifier. It is inferred from the start and stop arguments by default. axis (int): The axis in the result to store the samples. Relevant only if start or stop are array-like. By default ``0``, the samples will be along a new axis inserted at the beginning. Use ``-1`` to get an axis at the end. Returns: cupy.ndarray: The 1-D array of ranged values. .. seealso:: :func:`numpy.linspace` rz$linspace with num<0 is not supportedrr?Nrr')r()rr&r(r)Fr+) ValueErrorrisscalarr: isinstancendarrayr r asarrayr-rrrreshapendimrmoveaxisr1r2r3r4)rrr5r6r7r axisr9 scalar_start scalar_stopr8deltar!rs rlinspacerIzs`@ Qww?@@@ (377SC=''L-%%KL LtS(GUKKK .5$,// .EK4D4K4KL''#-E ,4.. ,4:?c3I3I<%%+D %uSzz 2 2B } 5LE *cV2 & & &C#sCr**** ++edUZ// 0 0C  Qwws{ 8DAI    3JC+CC*CCU||Ek5LCC!GGB qyymCD)) udl++! 3C    **U* ' 'CDy r$@ct|||||}|tj||Stj|jdvr|tj}tj|||S)a2Returns an array with evenly-spaced values on a log-scale. Instead of specifying the step width like :func:`cupy.arange`, this function requires the total number of elements specified. Args: start: Start of the interval. stop: End of the interval. num: Number of elements. endpoint (bool): If ``True``, the stop value is included as the last element. Otherwise, the stop value is omitted. base (float): Base of the log space. The step sizes between the elements on a log-scale are the same as ``base``. dtype: Data type specifier. It is inferred from the start and stop arguments by default. axis (int): The axis in the result to store the samples. Relevant only if start or stop are array-like. By default ``0``, the samples will be along a new axis inserted at the beginning. Use ``-1`` to get an axis at the end. Returns: cupy.ndarray: The 1-D array of ranged values. .. seealso:: :func:`numpy.logspace` )r5r6rENiu)rIrpowerrr r r4float64)rrr5r6baser rEys rlogspacerQs6 #tDDDA }{4### z%%% HHT\ " " ;tQ   & &u - --rc|dd}t|dd}t|dd}|r5tdt |d|d vrt d |D]E}|jd krt d t|tj st d Ft|d krt |Sg}t|D]o\}}|dkr |dkrd }n|dkr |d krd}n|}d|ztdfzdt||d zz zz} | || p|r|} nt tj|} |r?   a &'''' 9 .788  /s9~~&& / /A$Q<,,..IaLL rc&eZdZdZddZdZdZdS)nd_gridaConstruct a multi-dimensional "meshgrid". ``grid = nd_grid()`` creates an instance which will return a mesh-grid when indexed. The dimension and number of the output arrays are equal to the number of indexing dimensions. If the step length is not a complex number, then the stop is not inclusive. However, if the step length is a **complex number** (e.g. 5j), then the integer part of its magnitude is interpreted as specifying the number of points to create between the start and stop values, where the stop value **is inclusive**. If instantiated with an argument of ``sparse=True``, the mesh-grid is open (or not fleshed out) so that only one-dimension of each returned argument is greater than 1. Args: sparse (bool, optional): Whether the grid is sparse or not. Default is False. .. seealso:: :data:`numpy.mgrid` and :data:`numpy.ogrid` Fc||_dSrWrU)selfrUs r__init__znd_grid.__init__bs  rc t|tr|j}|j}|j}|d}t|t rmt |}t|}|dkr|j|z t|dz z }|j|z}tj d|dt|z|zStj |||Sg}t}tt|D] }||j}||j}|d}|d}t|t r7| tt |t}nH| ttj||j|z |dzz t|ts5t|ts t||jtrt} |jr,dt#||ft|zD} ntj||} tt|D]}||j}||j}|d}|d}t|t rEtt |}|dkr#||j|z t|dz z }| ||z|z| |<|jr|tjgt|z} tt|D]B}tdd| |<| |t)| | |<tj| |<C| S)Nrrr<c@g|]\}}tj||S)r)rr#)r _x_ts r z'nd_grid.__getitem__..s;???b"+b+++???r)r?r^rrrcomplexabsrrrr#rar\r_mathrrUzipindicesnewaxistuple) rokeyrrrlengthr r"knnslobjs r __getitem__znd_grid.__getitem__esB c5 ! ! 68D8DIE}$(( 64yyT199Hu,dQh?Dx${1fa77$>FF{5$555s3xx  Aq6;DFLE}|$(( J CD NN+++  3q6;#64#:"FGGHHJJJ4'' ue,, s1v{E22  ; )?? #D3&3t99*< = =???BBdC((Bs4yy!! + +Aq6;DFLE}|$(( C3t99~~199FK%/5??BDUT\E)BqEE ; (\NSYY.E3t99%% ( ( t,,a1eEll+1<a rcdS)Nr)ros r__len__znd_grid.__len__sqrN)F)__name__ __module__ __qualname____doc__rprrrrrrlrlIsQ0;;;zrrlrn cupy_arange)zbb->bzBB->Bzhh->hzHH->Hzii->izII->Izll->lzLL->Lzqq->qzQQ->Qzee->ezff->fdd->d)zFF->Fzout0 = in0 + float(i) * in1)zDD->Dzout0 = in0 + double(i) * in1zout0 = in0 + i * in1 cupy_linspace)r)zddd->dzout0 = in0 + i * in1 / in2)NrN)r$TFN)r$TFNr)r$TrJNr)rxrrrr#r:rIrQrjobjectrlmgridogrid create_ufuncrr0r/rrrrs| 4444nBG7777tGK^^^^BCG#.#.#.#.LEEEPZZZZZfZZZz ut#"0  %%$ /E. ""r