angle¶

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

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

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

参数¶

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

name (str，可选） - 操作的名称(可选，默认值为None）。更多信息请参见 Name。

返回¶

输出实数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.numpy())
# [[-2.-2.j -2.-1.j -2.+0.j -2.+1.j]
#  [-1.-2.j -1.-1.j -1.+0.j -1.+1.j]
#  [ 0.-2.j  0.-1.j  0.+0.j  0.+1.j]
#  [ 1.-2.j  1.-1.j  1.+0.j  1.+1.j]]

theta = paddle.angle(z)
print(theta.numpy())
# [[-2.3561945 -2.6779451  3.1415927  2.6779451]
#  [-2.0344439 -2.3561945  3.1415927  2.3561945]
#  [-1.5707964 -1.5707964  0.         1.5707964]
#  [-1.1071488 -0.7853982  0.         0.7853982]]

         