jax.Array.nonzero

jax.Array.nonzero#

abstract Array.nonzero(*, fill_value=None, size=None)[source]#

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

有关完整文档,请参考 jax.numpy.nonzero()

参数:
  • self (Array)

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

  • size (int | None)

返回值类型:

元组[数组, …]