mindspore.ops.float_power
- mindspore.ops.float_power(input, exponent)[源代码]
逐元素计算以第一个输入为底,第二个输入为指数。如果输入为实数类型,则转换为mindspore.float64计算。
说明
目前不支持复数运算。
- 参数:
input (Union[Tensor, Number]) - 第一个输入。
exponent (Union[Tensor, Number]) - 第二个输入。如果 input 是Number,则该参数必须是Tensor类型。
- 返回:
类型为mindspore.float64的tensor。
- 支持平台:
GPU
CPU
样例:
>>> import mindspore >>> # case 1: When exponent is scalar: >>> input = mindspore.tensor([-1.5, 0., 2.]) >>> output = mindspore.ops.float_power(input, 2) >>> print(output) [2.25 0. 4. ] >>> >>> # case 2: When exponent is a tensor: >>> exponent = mindspore.tensor([0, 1, 2]) >>> output = mindspore.ops.float_power(input, exponent) >>> print(output) [1. 0. 4.]