`iFddlZddlmZejddddZdZdS)N)_corezbool cond, T valuezT yzif (cond) y = valuecupy_piecewise_kernelctj|r|g}t|}t|}||kr tj|j|j}n|dz|krQ|d}|dd}t |rtdtj|j||j}n'td |||dzt||D]h\}}t |rtdt|tj r||j}t|||i|S)aEvaluate a piecewise-defined function. Args: x (cupy.ndarray): input domain condlist (list of cupy.ndarray): Each boolean array/ scalar corresponds to a function in funclist. Length of funclist is equal to that of condlist. If one extra function is given, it is used as the default value when the otherwise condition is met funclist (list of scalars): list of scalar functions. Returns: cupy.ndarray: the scalar values in funclist on portions of x defined by condlist. .. warning:: This function currently doesn't support callable functions, args and kw parameters. .. seealso:: :func:`numpy.piecewise` Nz.Callable functions are not supported currentlyz> B%@BB Biqw//))/'A+)N)NPP Px22.. 4 D>> B%@BB B dDL ) ) (;;qw''D 4---- J)rrElementwiseKernelrr!r"r r%sY )%)  00000r"