jax.numpy.pow

jax.numpy.pow(x1, x2, /)

第一个数组的元素按第二个数组的元素逐个求幂。

LAX 后端实现的 numpy.power()。

原始文档字符串如下。

将 x1 中的每个基数提升到 x2 中位置对应的幂。x1 和 x2 必须能够广播到相同的形状。

一个整数类型提升到负整数次幂将引发一个 ValueError。

负数值提升到非整数值将返回 nan 。要得到复数结果，请将输入转换为复数，或指定 dtype 为 complex （见下例）。

参数:

• x1 (array_like) – 基础。

• x2 (array_like) – 指数。如果 x1.shape != x2.shape，它们必须能够广播到一个共同的形状（这将成为输出的形状）。

返回:

y – x1 中的基数提升到 x2 中的指数。如果 x1 和 x2 都是标量，则这是一个标量。

返回类型:

ndarray