cupy.nonzero

cupy.nonzero(a)

返回非零元素的索引。

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

参数:

a (cupy.ndarray) – 数组

返回值:

非零元素的索引。

返回类型:

数组元组

警告

此函数可能会同步设备。

另请参阅

numpy.nonzero()