jax.lax.ne

jax.lax.ne(x, y)

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

此函数直接降级到具有 comparison_direction=NE 的 stablehlo.compare 操作，并根据输入 dtype 设置 compare_type。

参数:

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

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

返回:

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

返回类型:

数组

另请参阅

• jax.numpy.not_equal()：此 API 的 NumPy 包装器，也可通过 JAX 数组上的 x != y 运算符访问。

• jax.lax.eq(): 逐元素相等

• jax.lax.ge(): 逐元素大于等于

• jax.lax.gt(): 逐元素大于

• jax.lax.le(): 逐元素小于等于

• jax.lax.lt(): 逐元素小于