torch.polar

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

Создаёт комплексный тензор, элементы которого являются декартовыми координатами, соответствующими полярным координатам с абсолютным значением 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 бесконечно.

Параметры:

• abs (Тензор) – Абсолютное значение комплексного тензора. Должно иметь тип float или double.
• angle (Тензор) – Угол комплексного тензора. Должен иметь тот же тип данных, что и abs.

Именованные аргументы:

out (Тензор) – Если входные данные имеют тип torch.float32, должен иметь тип torch.complex64. Если входные данные имеют тип torch.float64, должен иметь тип torch.complex128.

Пример:

>>> 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)