torch.view_as_complex

torch.view_as_complex(input) → Tensor

Возвращает представление input как комплексного тензора. Для входного комплексного тензора размерности size $m1, m2, \dots, mi, 2$ эта функция возвращает новый комплексный тензор размерности size $m1, m2, \dots, mi$, где последняя размерность входного тензора ожидается, чтобы представлять вещественную и мнимую компоненты комплексных чисел.

Предупреждение

view_as_complex() поддерживается только для тензоров с torch.dtype torch.float64 и torch.float32. Ожидается, что входной тензор будет иметь последнюю размерность равную size 2. Кроме того, тензор должен иметь stride 1 для своей последней размерности. Шаги всех других измерений должны быть чётными числами.

Параметры:

input (Тензор) – входной тензор.

Пример:

>>> x=torch.randn(4, 2)
>>> x
tensor([[ 1.6116, -0.5772],
        [-1.4606, -0.9120],
        [ 0.0786, -1.7497],
        [-0.6561, -1.6623]])
>>> torch.view_as_complex(x)
tensor([(1.6116-0.5772j), (-1.4606-0.9120j), (0.0786-1.7497j), (-0.6561-1.6623j)])