jax.lax.pow

jax.lax.pow(x, y)

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

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

参数:

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

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

返回:

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

返回类型:

数组

另请参阅

jax.lax.integer_pow()：逐元素求幂，其中 y 是静态整数。