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 是一个数组。