cupy.piecewise#

cupy.piecewise(x, condlist, funclist)[source]#

计算分段函数。

参数:

  • x (cupy.ndarray) – 输入域

  • condlist (list of cupy.ndarray) – 每个布尔数组/标量对应于 funclist 中的一个函数。funclist 的长度等于 condlist 的长度。如果给出一个额外的函数，当不满足其他条件时，该函数将用作默认值。

  • funclist (list of scalars) – 标量函数列表。

返回值:

由 condlist 定义的 x 部分上的 funclist 中的标量值。

返回值类型:

cupy.ndarray

警告

此函数目前不支持可调用函数、args 和 kw 参数。

另请参阅

numpy.piecewise()