jax.lax.pow

jax.lax.pow(x, y)

逐元素求幂：\(x^y\)。

此函数直接降低为 stablehlo.pow 操作，以及当参数数据类型不匹配时的 stablehlo.convert。

参数:

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

• y (ArrayLike) – 输入数组，表示指数值。必须具有整数、浮点或复数类型。如有必要，其 dtype 将被强制转换为 x.dtype 的 dtype。如果 x 和 y 都不是标量，则 x 和 y 必须具有相同的维度数并且是广播兼容的。

返回:

一个与 x 具有相同 dtype 的数组，包含逐元素幂。

返回类型:

Array

另请参阅

jax.lax.integer_pow()：y 是静态整数的逐元素幂。