jax.numpy.reciprocal

jax.numpy.reciprocal(x, /)

计算输入的逐元素倒数。

JAX 对 numpy.reciprocal 的实现。

倒数通过 1/x 计算。

参数:

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

返回:

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

返回类型:

Array

注意

对于整数输入，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)