torch.angle#

torch.angle(input:Tensor,*,out:Optional[Tensor])→Tensor#

Computes the element-wise angle (in radians) of the giveninput tensor.

$${\text{out}}_i=angle({\text{input}}_i)$$outi​=angle(inputi​)

Parameters

input (Tensor) – the input tensor.

Keyword Arguments

out (Tensor,optional) – the output tensor.

Note

Starting in PyTorch 1.8, angle returns pi for negative real numbers,zero for non-negative real numbers, and propagates NaNs. Previouslythe function would return zero for all real numbers and not propagatefloating-point NaNs.

Example:

>>>torch.angle(torch.tensor([-1+1j,-2+2j,3-3j]))*180/3.14159tensor([ 135.,  135,  -45])