jax.numpy.absolute

jax.numpy.absolute(x, /)

逐元素計算絕對值。

numpy.absolute 的 JAX 實作。

這與 jax.numpy.abs() 函數相同。

參數:

x (ArrayLike) – 輸入陣列

返回值:

包含 x 中每個元素的絕對值的類陣列物件，形狀與 x 相同。對於複數值輸入 \(a + ib\)，絕對值為 \(\sqrt{a^2+b^2}\)。

返回類型:

陣列

範例

>>> x1 = jnp.array([5, -2, 0, 12])
>>> jnp.absolute(x1)
Array([ 5,  2,  0, 12], dtype=int32)

>>> x2 = jnp.array([[ 8, -3, 1],[ 0, 9, -6]])
>>> jnp.absolute(x2)
Array([[8, 3, 1],
       [0, 9, 6]], dtype=int32)

>>> x3 = jnp.array([8 + 15j, 3 - 4j, -5 + 0j])
>>> jnp.absolute(x3)
Array([17.,  5.,  5.], dtype=float32)