Функция torch.view_as_real в PyTorch
Справочное руководство PyTorch torch
torch.view_as_realЭто функция в PyTorch, используемая для преобразования представления комплексного тензора в вещественный тензор. Она разбивает каждое комплексное значение (действительную и мнимую части) на два отдельных значения, и последнее измерение будет содержать два элемента.
Определение функции
torch.view_as_real(input)
Пример использования
Пример
import torch
# Создать комплексный тензор
x = torch.tensor([1+2j, 3+4j, 5+6j])
# Преобразовать в вещественное представление
y = torch.view_as_real(x)
print("Форма исходного комплексного тензора:", x.shape)
print("Форма вещественного тензора:", y.shape)
print("Вещественный тензор:")
print(y)
# Создать комплексный тензор
x = torch.tensor([1+2j, 3+4j, 5+6j])
# Преобразовать в вещественное представление
y = torch.view_as_real(x)
print("Форма исходного комплексного тензора:", x.shape)
print("Форма вещественного тензора:", y.shape)
print("Вещественный тензор:")
print(y)
Результат вывода:
原始复数张量形状: torch.Size([3])
实数张量形状: torch.Size([3, 2])
实数张量:
tensor([[1., 2.],
[3., 4.],
[5., 6.]])
Другие расширения