cupy.flatnonzero

cupy.flatnonzero(a)

返回在 a 的展平版本中非零的索引。

这等同于 a.ravel().nonzero()[0]。

参数:

a (cupy.ndarray) – 输入数组

返回:

输出数组，包含 a.ravel() 中非零元素的索引。

返回类型:

cupy.ndarray

警告

此功能可能会同步设备。

参见

numpy.flatnonzero()