jax.lax.integer_pow#

jax.lax.integer_pow(x, y)[source]#

Elementwise power: \(x^y\), where \(y\) is a static integer.

This will lower to a sequence of \(O[\log_2(y)]\) repetitions of stablehlo.multiply.

Parameters:

  • x (ArrayLike) – Input array giving the base value. Must have numerical dtype.

  • y (int) – Static scalar integer giving the exponent.

Returns:

An array of the same shape and dtype as x containing the elementwise power.

Return type:

Array

See also

jax.lax.pow(): Elementwise pwoer where y is an array.