jax.numpy.absolute

jax.numpy.absolute(x, /)

逐元素计算绝对值。

JAX implementation of 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)