jax.numpy.ceil

jax.numpy.ceil(x, /)

Round input to the nearest integer upwards.

JAX implementation of numpy.ceil.

Parameters:

x (ArrayLike) – input array or scalar. Must not have complex dtype.

Returns:

An array with same shape and dtype as x containing the values rounded to the nearest integer that is greater than or equal to the value itself.

Return type:

Array

See also

• jax.numpy.fix(): Rounds the input to the nearest interger towards zero.

• jax.numpy.trunc(): Rounds the input to the nearest interger towards zero.

• jax.numpy.floor(): Rounds the input down to the nearest integer.

Examples

>>> key = jax.random.key(1)
>>> x = jax.random.uniform(key, (3, 3), minval=-5, maxval=5)
>>> with jnp.printoptions(precision=2, suppress=True):
...     print(x)
[[-0.61  0.34 -0.54]
 [-0.62  3.97  0.59]
 [ 4.84  3.42 -1.14]]
>>> jnp.ceil(x)
Array([[-0.,  1., -0.],
       [-0.,  4.,  1.],
       [ 5.,  4., -1.]], dtype=float32)