jax.lax.reduce_or#

jax.lax.reduce_or(operand, axes)[源代码]#

计算一个或多个数组轴上元素的按位 OR。

参数:

  • operand (ArrayLike) – 用于计算缩减的数组。必须具有布尔或整数 dtype。

  • axes (Sequence[int]) – 指定要约简的轴的零个或多个唯一整数序列。每个条目必须满足 0 <= axis < operand.ndim。

返回:

一个与 operand 具有相同 dtype 的数组，其形状对应于 operand.shape 中移除 axes 后的维度。

返回类型:

Array

另请参阅