cupy.argwhere#

cupy.argwhere(a)[source]#

返回非零元素的索引。

返回一个形状为 (N, ndim) 的数组，其中包含非零元素的索引。这里 N 是非零元素的数量，ndim 是给定数组的维度。

参数:

a (cupy.ndarray) – 数组

返回值:

非零元素的索引。

返回类型:

cupy.ndarray

另请参阅

numpy.argwhere()