jax.numpy.less

jax.numpy.less(x, y, /)

返回 x < y 的逐元素真值。

numpy.less 的 JAX 实现。

参数：

• x (ArrayLike) – 输入数组或标量。

• y (ArrayLike) – 输入数组或标量。 x 和 y 必须具有相同的形状或广播兼容。

返回：

包含布尔值的数组。 如果 x < y 的元素，则为 True，否则为 False。

返回类型：

Array

另请参阅

• jax.numpy.greater(): 返回 x > y 的逐元素真值。

• jax.numpy.greater_equal()：返回逐元素比较的真值，判断 x >= y 是否成立。

• jax.numpy.less_equal()：返回逐元素比较的真值，判断 x <= y 是否成立。

示例

标量输入

>>> jnp.less(3, 7)
Array(True, dtype=bool, weak_type=True)

具有相同形状的输入

>>> x = jnp.array([5, 9, -3])
>>> y = jnp.array([1, 6, 4])
>>> jnp.less(x, y)
Array([False, False,  True], dtype=bool)

具有广播兼容性的输入

>>> x1 = jnp.array([[2, -4, 6, -8],
...                 [-1, 5, -3, 7]])
>>> y1 = jnp.array([0, 3, -5, 9])
>>> jnp.less(x1, y1)
Array([[False,  True, False,  True],
       [ True, False, False,  True]], dtype=bool)