jax.lax.abs

jax.lax.abs(x)

逐元素绝对值： \(|x|\)。

此函数直接降低到 stablehlo.abs 操作。

参数:

x (ArrayLike) – 输入数组。必须具有带符号整数、浮点数或复数 dtype。

返回:

与 x 具有相同形状的数组，包含逐元素的绝对值。 对于复数值输入，\(a + ib\)，abs(x) 返回 \(\sqrt{a^2+b^2}\)。

返回类型:

数组

另请参阅

• jax.numpy.abs()：更灵活的 NumPy 风格的 abs 实现。