cupy.flatnonzero#
- cupy.flatnonzero(a)[source]#
返回将 a 展平后非零元素的索引。
这等价于 a.ravel().nonzero()[0]。
- 参数：
a (cupy.ndarray) – 输入数组
- 返回值：
输出数组，包含 a.ravel() 中非零元素的索引。
- 返回类型：
警告
此函数可能会同步设备。
另请参阅
返回将 a 展平后非零元素的索引。
这等价于 a.ravel().nonzero()[0]。
a (cupy.ndarray) – 输入数组
输出数组，包含 a.ravel() 中非零元素的索引。
警告
此函数可能会同步设备。
另请参阅