`i RdZddlZd dZd dZd dZd dZd dZd d Zd d Zdd Z dS)z- Functions for acting on a axis of an array. Nctdg|jz}t|||||<|t|}|S)aTake a slice along axis 'axis' from 'a'. Parameters ---------- a : cupy.ndarray The array to be sliced. start, stop, step : int or None The slice parameters. axis : int, optional The axis of `a` to be sliced. Examples -------- >>> a = array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) >>> axis_slice(a, start=0, stop=1, axis=1) array([[1], [4], [7]]) >>> axis_slice(a, start=1, axis=0) array([[4, 5, 6], [7, 8, 9]]) Notes ----- The keyword arguments start, stop and step are used by calling slice(start, stop, step). This implies axis_slice() does not handle its arguments the exactly the same as indexing. To select a single index k, for example, use axis_slice(a, start=k, stop=k+1) In this case, the length of the axis 'axis' in the result will be 1; the trivial dimension is not removed. (Use cupy.squeeze() to remove trivial axes.) Nslicendimtuple)astartstopstepaxisa_slicebs r/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/scipy/signal/_arraytools.py axis_slicersCDT{{maf$G%t,,GDM %..A Hctdg|jz}t|||||<||t|<|S)aiTake a slice along axis 'axis' from 'a' and set it to 'b' in-place. Parameters ---------- a : numpy.ndarray The array to be sliced. b : cupy.ndarray The array to be assigned. start, stop, step : int or None The slice parameters. axis : int, optional The axis of `a` to be sliced. Examples -------- >>> a = array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) >>> b1 = array([[-1], [-4], [-7]]) >>> axis_assign(a, b1, start=0, stop=1, axis=1) array([[-1, 2, 3], [-4, 5, 6], [-7, 8, 9]]) Notes ----- The keyword arguments start, stop and step are used by calling slice(start, stop, step). This implies axis_assign() does not handle its arguments the exactly the same as indexing. To assign a single index k, for example, use axis_assign(a, start=k, stop=k+1) In this case, the length of the axis 'axis' in the result will be 1; the trivial dimension is not removed. (Use numpy.squeeze() to remove trivial axes.) This function works in-place and will modify the values contained in `a` Nr)r rr r r r rs r axis_assignr/sCHT{{maf$G%t,,GDMAeGnn Hrc&t|d|S)zeReverse the 1-D slices of `a` along axis `axis`. Returns axis_slice(a, step=-1, axis=axis). r)r r )r)r r s r axis_reverserYs abt , , ,,rcn|dkr|S||j|dz kr"td||j|dz fzt|dd|}t||dd|}t|d|}t|d|d z d|}tjd |z|z |d |z|z f| }|S) a Odd extension at the boundaries of an array Generate a new ndarray by making an odd extension of `x` along an axis. Parameters ---------- x : ndarray The array to be extended. n : int The number of elements by which to extend `x` at each end of the axis. axis : int, optional The axis along which to extend `x`. Default is -1. Examples -------- >>> from cupyx.scipy.signal._arraytools import odd_ext >>> a = cupy.array([[1, 2, 3, 4, 5], [0, 1, 4, 9, 16]]) >>> odd_ext(a, 2) array([[-1, 0, 1, 2, 3, 4, 5, 6, 7], [-4, -1, 0, 1, 4, 9, 16, 23, 28]]) XThe extension length n (%d) is too big. It must not exceed x.shape[axis]-1, which is %d.rr r r rr r r r r r r shape ValueErrorrcupy concatenate)xnr left_endleft_ext right_end right_extexts rodd_extr,as. 1uu174=1 Mqwt}q01233 3!114888H!112DAAAH1BT222I1Bq1uXBTJJJI  AL83Q MI57=A C C CC Jrc |dkr|S||j|dz kr"td||j|dz fzt||dd|}t|d|dz d|}tj|||f|}|S) a Even extension at the boundaries of an array Generate a new ndarray by making an even extension of `x` along an axis. Parameters ---------- x : ndarray The array to be extended. n : int The number of elements by which to extend `x` at each end of the axis. axis : int, optional The axis along which to extend `x`. Default is -1. Examples -------- >>> from cupyx.scipy.signal._arraytools import even_ext >>> a = cupy.array([[1, 2, 3, 4, 5], [0, 1, 4, 9, 16]]) >>> even_ext(a, 2) array([[ 3, 2, 1, 2, 3, 4, 5, 4, 3], [ 4, 1, 0, 1, 4, 9, 16, 9, 4]]) rrrrrrrrr )r%r&r r(r*r+s reven_extr.s0 1uu174=1 Mqwt}q01233 3!112DAAAH1Bq1uXBTJJJI  Ha3$ ? ? ?C Jrc|dkr|St|dd|}dg|jz}|||<tj||j}||z}t|d|}||z}tj|||f|} | S)a Constant extension at the boundaries of an array Generate a new ndarray that is a constant extension of `x` along an axis. The extension repeats the values at the first and last element of the axis. Parameters ---------- x : ndarray The array to be extended. n : int The number of elements by which to extend `x` at each end of the axis. axis : int, optional The axis along which to extend `x`. Default is -1. Examples -------- >>> from cupyx.scipy.signal._arraytools import const_ext >>> a = cupy.array([[1, 2, 3, 4, 5], [0, 1, 4, 9, 16]]) >>> const_ext(a, 2) array([[ 1, 1, 1, 2, 3, 4, 5, 5, 5], [ 0, 0, 0, 1, 4, 9, 16, 16, 16]]) rrrdtyperrr)rrr#onesr1r$) r%r&r r' ones_shaper2r(r)r*r+s r const_extr4s2 1uu!114888HqvJJt 9Zqw / / /DhH1BT222Iy I  Ha3$ ? ? ?C Jrc|dkr|St|j}|||<tj||j}tj|||f|}|S)a Zero padding at the boundaries of an array Generate a new ndarray that is a zero-padded extension of `x` along an axis. Parameters ---------- x : ndarray The array to be extended. n : int The number of elements by which to extend `x` at each end of the axis. axis : int, optional The axis along which to extend `x`. Default is -1. Examples -------- >>> from cupyx.scipy.signal._arraytools import zero_ext >>> a = cupy.array([[1, 2, 3, 4, 5], [0, 1, 4, 9, 16]]) >>> zero_ext(a, 2) array([[ 0, 0, 1, 2, 3, 4, 5, 0, 0], [ 0, 0, 0, 1, 4, 9, 16, 0, 0]]) rr0r)listr!r#zerosr1r$)r%r&r zeros_shaper7r+s rzero_extr9s`2 1uuqw--KK J{!' 2 2 2E  E1e,4 8 8 8C Jrc||jnt|}||jnt|}tj||j|j|S)a Create a view into the array with the given shape and strides. .. warning:: This function has to be used with extreme care, see notes. Parameters ---------- x : ndarray Array to create a new. shape : sequence of int, optional The shape of the new array. Defaults to ``x.shape``. strides : sequence of int, optional The strides of the new array. Defaults to ``x.strides``. Returns ------- view : ndarray Notes ----- ``as_strided`` creates a view into the array given the exact strides and shape. This means it manipulates the internal data structure of ndarray and, if done incorrectly, the array elements can point to invalid memory and can corrupt results or crash your program. N)r!r1memptrstrides)r!rr<r#ndarrayr1data)r%r!r<s r _as_stridedr?sX2}AGG%,,E"?aiigG <17167 D D DDr)NNNr)r)NN) __doc__r#rrrr,r.r4r9r?rrrBs % % % % P' ' ' ' T----####L!!!!H####LDDDDDDDr