jax.numpy.reciprocal

jax.numpy.reciprocal(x, /)

计算输入的逐元素倒数。

numpy.reciprocal 的 JAX 实现。

倒数计算方式为 1/x。

参数:

x (ArrayLike) – 输入数组或标量。

返回:

一个与 x 形状相同的数组，包含 x 中每个元素的倒数。

返回类型:

数组

注意

对于整数输入，np.reciprocal 返回四舍五入的整数输出，而 jnp.reciprocal 将整数输入提升为浮点数。

示例

>>> jnp.reciprocal(2)
Array(0.5, dtype=float32, weak_type=True)
>>> jnp.reciprocal(0.)
Array(inf, dtype=float32, weak_type=True)
>>> x = jnp.array([1, 5., 4.])
>>> jnp.reciprocal(x)
Array([1.  , 0.2 , 0.25], dtype=float32)