jax.numpy.fix

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

将输入四舍五入到最接近的整数，趋向于零。

JAX 实现的 numpy.fix()。

参数:

• x (ArrayLike) – 输入数组。

• out (None) – JAX 未使用。

返回:

一个与 x 形状和数据类型相同的数组，包含四舍五入后的值。

返回类型:

Array

参见

• 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)
[[-1.45  1.04 -0.72]
 [-2.69  1.74 -0.6 ]
 [-2.49 -2.23  2.68]]
>>> jnp.fix(x)
Array([[-1.,  1., -0.],
       [-2.,  1., -0.],
       [-2., -2.,  2.]], dtype=float32)