jax.lax.exp2

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

Elementwise base-2 指数：\(2^x\)。

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

参数:

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

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

返回值:

与 x 具有相同形状和数据类型的数组，包含逐元素的 base-2 指数。

返回类型:

数组

参见

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

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