jax.numpy.ceil

jax.numpy.ceil(x, /)

将输入向上舍入到最接近的整数。

JAX 对 numpy.ceil 的实现。

参数:

x (ArrayLike) – 输入数组或标量。不得具有复数数据类型。

返回:

一个与 x 形状和数据类型相同的数组，其中包含向上舍入到最接近的整数（大于或等于原始值）的值。

返回类型:

数组

另请参阅

• jax.numpy.fix()：将输入舍入到最接近零的整数。

• jax.numpy.trunc()：将输入舍入到最接近零的整数。

• jax.numpy.floor()：将输入向下舍入到最接近的整数。

示例

>>> 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)