cupy.extract

cupy.extract(condition, a)

返回满足某些条件的数组元素。

这等价于 np.compress(ravel(condition), ravel(arr))。如果 condition 是布尔值，np.extract 等价于 arr[condition]。

参数:

• condition (int or array_like) – 一个数组，其非零或True的条目指示要提取的数组元素。

• a (cupy.ndarray) – 与条件相同大小的输入数组。

返回:

从数组 arr 中获取条件为 True 的值，并将其排序为第一级数组。

返回类型:

cupy.ndarray

警告

此功能可能会同步设备。

参见

numpy.extract()