Функция torch.view_as_complex в PyTorch
Справочное руководство Pytorch torch
torch.view_as_complexЭто функция в PyTorch, используемая для преобразования представления вещественного тензора в комплексный тензор. Она преобразует последнее измерение из двух вещественных значений (действительной и мнимой частей) в одно комплексное значение.
Определение функции
torch.view_as_complex(input)
Пример использования
Пример
import torch
# Создать вещественный тензор; последнее измерение должно содержать 2 элемента
x = torch.randn(3, 2)
# Преобразовать в комплексное представление
y = torch.view_as_complex(x)
print("Форма исходного вещественного тензора:", x.shape)
print("Форма комплексного тензора:", y.shape)
print("Комплексный тензор:", y)
# Создать вещественный тензор; последнее измерение должно содержать 2 элемента
x = torch.randn(3, 2)
# Преобразовать в комплексное представление
y = torch.view_as_complex(x)
print("Форма исходного вещественного тензора:", x.shape)
print("Форма комплексного тензора:", y.shape)
print("Комплексный тензор:", y)
Результат вывода:
原始实数张量形状: torch.Size([3, 2]) 复数张量形状: torch.Size([3]) 复数张量: tensor([ 1.2345+0.5678j, -0.3456+1.2345j, 2.3456-0.9876j])
Другие расширения