jax.numpy.greater

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

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

JAX 对 numpy.greater 的实现。

参数:

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

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

返回:

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

返回类型:

数组

另请参阅

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

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

• jax.numpy.less_equal(): 返回 x <= y 的逐元素真值。

示例

标量输入

>>> jnp.greater(5, 2)
Array(True, dtype=bool, weak_type=True)

具有相同形状的输入

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

支持广播的输入

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