cupy.extract

cupy.extract(condition, a)

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

这等同于 np.compress(ravel(condition), ravel(arr))。如果 condition 是布尔类型，则 np.extract 等同于 arr[condition]。

参数:

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

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

返回值:

来自 arr 的一维数组，其中 condition 为 True。

返回类型:

cupy.ndarray

警告

此函数可能会同步设备。

另请参阅

numpy.extract()