jax.lax.pow#

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

Elementwise power:\(x^y\).

This function lowers directly to thestablehlo.pow operation, along withastablehlo.convert when the argument dtypes do not match.

Parameters:

• x (ArrayLike) – Input array giving the base value. Must have floating or complex type.

• y (ArrayLike) – Input array giving the exponent value. Must have integer, floating, orcomplex type. Its dtype will be cast to that ofx.dtype if necessary.If neitherx nory is a scalar, thenx andy must havethe same number of dimensions and be broadcast-compatible.

Returns:

An array of the same dtype asx containing the elementwise power.

Return type:

Array

See also

jax.lax.integer_pow(): Elementwise power wherey is a static integer.