jax.lax.floor

jax.lax.floor(x)

Elementwise floor: \(\left\lfloor x \right\rfloor\).

This function lowers directly to the stablehlo.floor operation.

Parameters:

x (ArrayLike) – input array. Must have floating-point type.

Returns:

Array of same shape and dtype as x, containing values rounded to the next integer toward negative infinity.

Return type:

Array

See also

• jax.lax.ceil(): round to the next integer toward positive infinity

• jax.lax.round(): round to the nearest integer