jax.numpy.fabs

jax.numpy.fabs(x, /)

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

JAX implementation of numpy.fabs.

参数:

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

返回:

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

返回类型:

Array

参见

• jax.numpy.absolute(): 计算输入的绝对值，包括复数数据类型。

• jax.numpy.abs(): 计算输入的绝对值，包括复数数据类型。

示例

对于整数输入：

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