mindspore.ops.angle

mindspore.ops.angle(input)

逐元素计算复数Tensor的辐角。 输入中的元素是形式为a+bj的复数，其中a是实部，b是虚部。此函数返回的参数的形式为 \(atan2(b, a)\) 。

参数：

• input (Tensor) - 输入Tensor。支持类型：complex64、complex128。

返回：

Tensor，类型为float32或float64，shape与输入相同。

异常：

• TypeError - 如果 input 不是Tensor。

• TypeError - 如果输入的数据类型不是complex64或complex128。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> from mindspore import Tensor, ops
>>> input = Tensor([-1.5 + 7.8j, 3 + 5.75j], mindspore.complex64)
>>> output = ops.angle(input)
>>> print(output)
[1.7607845 1.0899091]