jax.numpy.fabs

jax.numpy.fabs(x, /)

计算实值输入的逐元素绝对值。

numpy.fabs 的 JAX 实现。

参数:

x (ArrayLike) – 输入数组或标量。不能具有复数 dtype。

返回:

一个与 x 形状相同，dtype 为 float 的数组，包含逐元素的绝对值。

返回类型:

数组

另请参阅

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