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。
- 返回类型:
参见
numpy.select()