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相同,数据类型为 Tensor 与 power 中精度较高的类型。
- 异常:
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.]