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, …]