jax.lax.le

jax.lax.le(x, y)

逐元素小于或等于：\(x \leq y\)。

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

参数:

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

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

返回:

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

返回类型:

Array

参见

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

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

• jax.lax.ne()：逐元素不相等

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

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

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