mindspore.mint.pow
- mindspore.mint.pow(input, exponent)[源代码]
计算 input 中每个元素的 exponent 次幂。
当 exponent 是Tensor时, input 和 exponent 的shape必须是可广播的。
\[out_{i} = input_{i} ^{ exponent_{i}}\]- 参数:
- 返回:
Tensor,shape与广播后的shape相同,数据类型为两个输入中精度较高的类型。
- 支持平台:
Ascend
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, mint >>> input = Tensor(np.array([1.0, 2.0, 4.0]), mindspore.float32) >>> exponent = 3.0 >>> output = mint.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 = mint.pow(input, exponent) >>> print(output) [ 1. 16. 64.]