cupy.select

cupy.select(condlist, choicelist, default=0)

根据条件从 choicelist 中的元素中返回一个数组。

参数：

• condlist (list of bool arrays) – 决定从 choicelist 中的哪个数组中获取输出元素的条件列表。当多个条件满足时，使用在 condlist 中遇到的第一个条件。

• choicelist (list of cupy.ndarray) – 用于获取输出元素的数组列表。它的长度必须与 condlist 相同。

• default (scalar) – 如果提供，当所有条件都评估为 False 时，将填充插入到 output 中的元素。默认值为 0。

返回：

位置 m 的输出是 choicelist 中对应数组的第 m 个元素，其中 condlist 中相应数组的第 m 个元素为 True。

返回类型：

cupy.ndarray

另请参阅

numpy.select()