torch.angle

torch.angle(input: 張量, *, out: Optional[張量]) → 張量

計算給定 input 張量的逐元素角度（以弧度為單位）。

$$\text{out}_{i} = angle(\text{input}_{i})$$

引數

input (張量) – 輸入張量。

關鍵字引數

out (張量, 可選) – 輸出張量。

注意

從 PyTorch 1.8 開始，angle 對於負實數返回 pi，對於非負實數返回零，並傳播 NaN。以前該函式對所有實數返回零，並且不傳播浮點 NaN。

示例

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