Python PyTorch polar用法及代碼示例

本文簡要介紹python語言中 torch.polar 的用法。

用法:

torch.polar(abs, angle, *, out=None) → Tensor

參數：

• abs(Tensor) -複張量的絕對值。必須是浮點數或雙精度數。

• angle(Tensor) -複張量的角度。必須與 abs 具有相同的 dtype。

關鍵字參數：

out(Tensor) -如果輸入是 torch.float32 ，則必須是 torch.complex64 。如果輸入是 torch.float64 ，則必須是 torch.complex128 。

構造一個複數張量，其元素是笛卡爾坐標，對應於具有絕對值 abs 和角度 angle 的極坐標。

\text{out} = \text{abs} \cdot \cos(\text{angle}) + \text{abs} \cdot \sin(\text{angle}) \cdot j

注意

torch.polar 類似於 std::polar 並且不會像 Python 的 cmath.polar 和 SciPy 的 linalg.polar 那樣計算複雜張量的極分解。如果 abs 為負數或 NaN，或者 angle 為無限，則此函數的行為未定義。

例子：

>>> import numpy as np
>>> abs = torch.tensor([1, 2], dtype=torch.float64)
>>> angle = torch.tensor([np.pi / 2, 5 * np.pi / 4], dtype=torch.float64)
>>> z = torch.polar(abs, angle)
>>> z
tensor([(0.0000+1.0000j), (-1.4142-1.4142j)], dtype=torch.complex128)