jax.lax.exp2

jax.lax.exp2(x, accuracy=None)

逐元素底数为 2 的指数函数：\(2^x\)。

此函数是根据 stablehlo.exponential 和 stablehlo.multiply 操作实现的。

参数:

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

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

返回:

与 x 形状和数据类型相同的数组，包含逐元素底数为 2 的指数。

返回类型:

Array (数组)

另请参阅

• jax.lax.exp(): 逐元素指数函数: \(e^x\).

• jax.lax.log(): 逐元素自然对数: \(\mathrm{log}(x)\).