jax.Array.nonzero

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

傳回陣列中非零元素的索引。

請參閱 jax.numpy.nonzero() 以取得完整文件。

參數:

• self (Array)

• fill_value (None | ArrayLike | tuple[ArrayLike, ...])

• size (int | None)

傳回型別:

tuple[Array, …]