jax.numpy.fabs

jax.numpy.fabs(x, /)

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

JAX 实现 numpy.fabs.

参数:

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

返回:

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

返回类型:

Array

另请参见

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