jax.lax.expm1#

jax.lax.expm1(x, accuracy=None)[source]#

逐元素运算 \(e^{x} - 1\).

此函数直接降级到 stablehlo.exponential_minus_one 操作。与朴素表达式 lax.exp(x) - 1 相比，对于接近零的 x，它更准确。

参数:

  • x (ArrayLike) – 输入数组。 必须具有浮点型或复数类型。

  • accuracy – 可选的 lax.Tolerance 或 lax.AccuracyMode 对象，用于根据请求的精度选择op的实现。如果实现无法满足请求的容差，编译器将返回错误。如果指定了模式并且没有多个实现可用，则将使用默认实现。

返回:

与 x 具有相同形状和 dtype 的数组，包含逐元素的指数减 1。

返回类型:

数组

另请参阅