jax.numpy.greater_equal

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

傳回元素級別的 x >= y 真值。

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)