jax.numpy.fabs

jax.numpy.fabs(x, /)

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

numpy.fabs 的 JAX 实现。

参数:

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

返回:

一个与 x 形状相同、数据类型为浮点数的数组，包含逐元素的绝对值。

返回类型:

数组

另请参阅

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