angle

paddle. angle ( x, name=None ) [源代码]

逐元素计算复数的相位角。对于非负实数,相位角为 0,而对于负实数,相位角为 \(\pi\)

\[angle(x) = arctan2(x.imag, x.real)\]

参数

  • x (Tensor) - 输入的 Tensor,数据类型为:complex64, complex128 或 float32, float64。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

输出实数 Tensor,与 x 的数值精度一致。

代码示例

import paddle

x = paddle.to_tensor([-2, -1, 0, 1]).unsqueeze(-1).astype('float32')
y = paddle.to_tensor([-2, -1, 0, 1]).astype('float32')
z = x + 1j * y
print(z)
# Tensor(shape=[4, 4], dtype=complex64, place=Place(cpu), stop_gradient=True,
#        [[(-2-2j), (-2-1j), (-2+0j), (-2+1j)],
#         [(-1-2j), (-1-1j), (-1+0j), (-1+1j)],
#         [-2j    , -1j    ,  0j    ,  1j    ],
#         [ (1-2j),  (1-1j),  (1+0j),  (1+1j)]])

theta = paddle.angle(z)
print(theta)
# Tensor(shape=[4, 4], dtype=float32, place=Place(cpu), stop_gradient=True,
#        [[-2.35619450, -2.67794514,  3.14159274,  2.67794514],
#         [-2.03444386, -2.35619450,  3.14159274,  2.35619450],
#         [-1.57079637, -1.57079637,  0.        ,  1.57079637],
#         [-1.10714877, -0.78539819,  0.        ,  0.78539819]])