jax.numpy.ceil

jax.numpy.ceil(x, /)

將輸入向上捨入到最接近的整數。

NumPy ceil 的 JAX 實作。

參數:

x (ArrayLike) – 輸入陣列或純量。不得為複數 dtype。

回傳:

一個與 x 具有相同形狀和 dtype 的陣列，其中包含捨入到最接近的整數值，且該整數值大於或等於該值本身。

回傳類型:

陣列

另請參閱

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