jax.numpy.ceil

jax.numpy.ceil(x, /)

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

numpy.ceil 的 JAX 实现。

参数:

x (ArrayLike) – 输入数组或标量。不能有复数 dtype。

返回:

与 x 具有相同形状和 dtype 的数组，其中包含四舍五入到最接近的整数的值，该整数大于或等于该值本身。

返回类型:

Array

参见

• 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)
[[ 2.55 -1.87 -3.76]
 [ 0.48  3.85 -1.94]
 [ 3.2   4.56 -1.43]]
>>> jnp.ceil(x)
Array([[ 3., -1., -3.],
       [ 1.,  4., -1.],
       [ 4.,  5., -1.]], dtype=float32)