jax.numpy.floor

jax.numpy.floor(x, /)

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

numpy.floor 的 JAX 實作。

參數:

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

返回:

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

返回類型:

Array

另請參閱

• jax.numpy.fix()：將輸入值朝零方向捨入到最接近的整數。

• jax.numpy.trunc()：將輸入值朝零方向捨入到最接近的整數。

• jax.numpy.ceil()：將輸入值向上捨入到最接近的整數。

範例

>>> key = jax.random.key(42)
>>> x = jax.random.uniform(key, (3, 3), minval=-5, maxval=5)
>>> with jnp.printoptions(precision=2, suppress=True):
...     print(x)
[[-0.11  1.8   1.16]
 [ 0.61 -0.49  0.86]
 [-4.25  2.75  1.99]]
>>> jnp.floor(x)
Array([[-1.,  1.,  1.],
       [ 0., -1.,  0.],
       [-5.,  2.,  1.]], dtype=float32)