jax.numpy.greater_equal

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

返回元素级别的真值，表示 x >= y。

NumPy numpy.greater_equal 的 JAX 实现。

参数：

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

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

返回值：

一个包含布尔值的数组。如果 x >= y 的元素为真，则为 True，否则为 False。

返回类型：

Array

参见

• jax.numpy.less_equal()：返回元素级别的真值，表示 x <= y。

• jax.numpy.greater()：返回元素级别的真值，表示 x > y。

• jax.numpy.less()：返回元素级别的真值，表示 x < y。

示例

标量输入

>>> jnp.greater_equal(4, 7)
Array(False, dtype=bool, weak_type=True)

具有相同形状的输入

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

具有广播兼容性的输入

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