jax.lax.ceil

jax.lax.ceil(x)

Elementwise ceiling: \(\left\lceil x \right\rceil\).

This function lowers directly to the stablehlo.ceil 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 positive infinity.

Return type:

Array

See also

• jax.lax.floor(): round to the next integer toward negative infinity

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