jax.numpy.absolute

jax.numpy.absolute(x, /)

逐元素计算绝对值。

JAX 实现 numpy.absolute。

此函数与 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)