jax.lax.lt

jax.lax.lt(x, y)

Elementwise less-than: \(x < y\).

This function lowers directly to the stablehlo.compare operation with comparison_direction=LT and compare_type set according to the input dtype.

Parameters:

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

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

Returns:

A boolean array of shape lax.broadcast_shapes(x.shape, y.shape) containing the elementwise less-than comparison.

Return type:

Array

See also

• jax.numpy.less(): NumPy wrapper for this API, also accessible via the x < y operator on JAX arrays.

• jax.lax.eq(): elementwise equal

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

• jax.lax.ge(): elementwise greater-than-or-equal

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

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