jax.lax.ne#

jax.lax.ne(x, y)[源代码]#

逐元素不等：\(x \neq y\)。

此函数直接降低到 stablehlo.compare 操作，其中 comparison_direction=NE 且 compare_type 根据输入 dtype 设置。

参数:

  • x (ArrayLike) – 输入数组。必须具有匹配的 dtypes。如果都不是标量，则 x 和 y 必须具有相同的维度数并且是广播兼容的。

  • y (ArrayLike) – 输入数组。必须具有匹配的 dtypes。如果都不是标量，则 x 和 y 必须具有相同的维度数并且是广播兼容的。

返回:

形状为 lax.broadcast_shapes(x.shape, y.shape) 的布尔数组，包含逐元素不等比较。

返回类型:

Array

另请参阅