mlx.core.power

power(a: scalar | array, b: scalar | array, /, *, stream: None | Stream |Device = None) → array

逐元素幂运算。

按照 numpy 风格的广播语义，将数组 a 的元素提升为数组 b 中元素对应的幂次。输入数组 a 和 b 都可以是标量。

参数：

• a (array) – 输入数组或标量。

• b (array) – 输入数组或标量。

返回值：

将 a 中的底数提升到 b 中的幂次的结果。

返回类型：

array