jax.lax.integer_pow

jax.lax.integer_pow(x, y)

逐元素幂运算：\(x^y\)，其中 \(y\) 是静态整数。

这将降低到 \(O[\log_2(y)]\) 次重复的 stablehlo.multiply 序列。

参数:

• x (ArrayLike) – 输入数组，给出底数值。必须具有数值 dtype。

• y (int) – 静态标量整数，给出指数。

返回:

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

返回类型:

Array

另请参阅

jax.lax.pow()：逐元素幂运算，其中 y 是一个数组。