jax.lax.abs

jax.lax.abs(x)

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

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

参数:

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

返回值:

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

返回类型:

Array

另请参阅

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