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 ofstablehlo.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 asx containing the elementwise power.

Return type:

Array

See also

jax.lax.pow(): Elementwise power wherey is an array.