mindspore.Tensor.pow

mindspore.Tensor.pow(power)[源代码]

计算Tensor中每个元素的 power 次幂。

\[out_{i} = x_{i} ^{ y_{i}}\]

Note

  • Tensor和 power 遵循 隐式类型转换规则 ,使数据类型保持一致。

  • 当前的Tensor和 power 的数据类型不能同时是bool,并保证其shape可以广播。

参数:
  • power (Union[Tensor, number.Number, bool]) - 幂值,是一个number.Number或bool值,或数据类型为number或bool_的Tensor。

返回:

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

异常:
  • TypeError - power 不是Tensor、number.Number或bool。

  • ValueError - 当Tensor和 power 都为Tensor时,它们的shape不相同。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([1.0, 2.0, 4.0]), mindspore.float32)
>>> y = 3.0
>>> output = x.pow(y)
>>> print(output)
[ 1.  8. 64.]
>>>
>>> x = Tensor(np.array([1.0, 2.0, 4.0]), mindspore.float32)
>>> y = Tensor(np.array([2.0, 4.0, 3.0]), mindspore.float32)
>>> output = x.pow(y)
>>> print(output)
[ 1. 16. 64.]