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)