jax.numpy.absolute#

jax.numpy.absolute(x, /)[source]#

逐元素计算绝对值。

JAX 实现的 numpy.absolute。

这是与 jax.numpy.abs() 相同的函数。

参数:

x (ArrayLike) – 输入数组

返回值:

一个类数组对象，包含 x 中每个元素的绝对值，形状与 x 相同。对于复数值输入 $a+ib$，绝对值为 $\sqrt{a^2+b^2}$。

返回类型:

Array

示例

>>> 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)