jax.lax.integer_pow

Contents

jax.lax.integer_pow#

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

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

Parameters:
  • x (jax.typing.ArrayLike)

  • y (int)

Return type:

Array