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.]