jax.numpy.ceil#
- jax.numpy.ceil(x, /)[source]#
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:
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)