jax.Array.all

abstract Array.all(axis=None, out=None, keepdims=False, *, where=None)

测试给定轴上的所有数组元素是否都计算为 True。

完整文档请参考jax.numpy.all()。

参数:

• self (Array)

• axis (reductions.Axis)

• out (None)

• keepdims (bool)

• where (ArrayLike | None)

返回类型:

Array