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) 返回值类型: 元组[数组, …]