jax.numpy.fix

jax.numpy.fix(x, out=None)

將輸入四捨五入到最接近零的整數。

numpy.fix() 的 JAX 實作。

參數:

• x (ArrayLike) – 輸入陣列。

• out (None) – JAX 未使用。

回傳值:

與 x 具有相同形狀和 dtype 的陣列，包含四捨五入的值。

回傳類型:

陣列

另請參閱

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

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

• jax.numpy.floor()：將輸入向下四捨五入到最接近的整數。

範例

>>> key = jax.random.key(0)
>>> x = jax.random.uniform(key, (3, 3), minval=-5, maxval=5)
>>> with jnp.printoptions(precision=2, suppress=True):
...     print(x)
[[ 4.48  4.79 -1.68]
 [-0.31  0.7  -3.34]
 [-1.9   1.89  2.47]]
>>> jnp.fix(x)
Array([[ 4.,  4., -1.],
       [-0.,  0., -3.],
       [-1.,  1.,  2.]], dtype=float32)