cupy.nonzero

cupy.nonzero(a)

返回非零元素的索引。

返回一个数组元组，a 的每个维度对应一个数组，包含该维度中非零元素的索引。

参数:

a (cupy.ndarray) – 数组

返回:

非零元素的索引。

返回类型:

tuple of arrays

警告

此功能可能会同步设备。

参见

numpy.nonzero()