jax.numpy.fabs

jax.numpy.fabs(x, /)

計算實數值輸入的逐元素絕對值。

numpy.fabs 的 JAX 實作。

參數:

x (ArrayLike) – 輸入陣列或純量。不得具有複數 dtype。

返回:

一個與 x 形狀相同且 dtype 為 float 的陣列，包含逐元素的絕對值。

返回類型:

Array

參見

• jax.numpy.absolute()：計算輸入的絕對值，包括複數 dtype。

• jax.numpy.abs()：計算輸入的絕對值，包括複數 dtype。

範例

對於整數輸入

>>> x = jnp.array([-5, -9, 1, 10, 15])
>>> jnp.fabs(x)
Array([ 5.,  9.,  1., 10., 15.], dtype=float32)

對於浮點數型別輸入

>>> x1 = jnp.array([-1.342, 5.649, 3.927])
>>> jnp.fabs(x1)
Array([1.342, 5.649, 3.927], dtype=float32)

對於布林輸入

>>> x2 = jnp.array([True, False])
>>> jnp.fabs(x2)
Array([1., 0.], dtype=float32)