jax.lax.ge#

jax.lax.ge(x,y)[source]#

Elementwise greater-than-or-equals:\(x \geq y\).

This function lowers directly to thestablehlo.compare operationwithcomparison_direction=GE andcompare_type set accordingto the input dtype.

Parameters:

• x (ArrayLike) – Input arrays. Must have matching non-complex dtypes. If neither isa scalar,x andy must have the same number of dimensions andbe broadcast compatible.

• y (ArrayLike) – Input arrays. Must have matching non-complex dtypes. If neither isa scalar,x andy must have the same number of dimensions andbe broadcast compatible.

Returns:

A boolean array of shapelax.broadcast_shapes(x.shape,y.shape)containing the elementwise greater-than-or-equal comparison.

Return type:

Array

See also

• jax.numpy.greater_equal(): NumPy wrapper for this API, alsoaccessible via thex>=y operator on JAX arrays.

• jax.lax.eq(): elementwise equal

• jax.lax.ne(): elementwise not-equal

• jax.lax.gt(): elementwise greater-than

• jax.lax.le(): elementwise less-than-or-equal

• jax.lax.lt(): elementwise less-than