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]