o i@s*ddlZddlmZddZdddZdS)N)_corec Cspzt|}Wn ty|f}Ynwt|}|dt|kr.t|tjr.tj|d|dStj|d|d}||jkrDd|j||}tddt |j |D}|j dkr^tj ||j d Sg}g}t |j |D]6\}} | dkr{||||qh|dkr|||| qh|d|||| ||qhtj ||j d } | j rt||| | |S) a Construct an array by repeating A the number of times given by reps. Args: A (cupy.ndarray): Array to transform. reps (int or tuple): The number of repeats. Returns: cupy.ndarray: Transformed array with repeats. .. seealso:: :func:`numpy.tile` T)copyndminF)rcss|] \}}||VqdSN).0strrV/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/cupy/_manipulation/tiling.py !sztile..r)dtype)tuple TypeErrorlencount isinstancecupyndarrayarrayndimzipshapesizeemptyr appendrelementwise_copyreshape) Arepstupdc shape_outc_shape ret_shapedim_innrepretrrr tiles>              r)cCs |||S)a+Repeat arrays along an axis. Args: a (cupy.ndarray): Array to transform. repeats (int, list or tuple): The number of repeats. axis (int): The axis to repeat. Returns: cupy.ndarray: Transformed array with repeats. .. seealso:: :func:`numpy.repeat` )repeat)arepeatsaxisrrr r*8s r*r)rrr)r*rrrr s 3