jax.Array.nonzero

abstract Array.nonzero(*, fill_value=None, size=None)

返回数组中非零元素的索引。

有关完整文档，请参阅 jax.numpy.nonzero()。

参数：

• self ( Array )

• fill_value (None | 类数组 | tuple[类数组, ...])

• size ( int | None )

返回类型：

tuple[Array, …]