mindspore.ops.pow

查看源文件
mindspore.ops.pow(input, exponent)[源代码]

计算 input 中每个元素的 exponent 次幂。

exponent 是Tensor时, inputexponent 的shape必须是可广播的。

\[out_{i} = input_{i} ^{ exponent_{i}}\]
参数:
  • input (Union[Tensor, Number]) - 第一个输入,是一个Number值或数据类型为 numberbool_ 的Tensor。

  • exponent (Union[Tensor, Number]) - 第二个输入,是一个Number值或数据类型为 numberbool_ 的Tensor。

返回:

Tensor,shape与广播后的shape相同,数据类型为两个输入中精度较高的类型。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input = Tensor(np.array([1.0, 2.0, 4.0]), mindspore.float32)
>>> exponent = 3.0
>>> output = ops.pow(input, exponent)
>>> print(output)
[ 1.  8. 64.]
>>>
>>> input = Tensor(np.array([1.0, 2.0, 4.0]), mindspore.float32)
>>> exponent = Tensor(np.array([2.0, 4.0, 3.0]), mindspore.float32)
>>> output = ops.pow(input, exponent)
>>> print(output)
[ 1. 16. 64.]