Функция real
Функция real возвращает действительную часть комплексного тензора.
Она принимает на вход тензор с комплексными числами и возвращает новый тензор,
содержащий только действительные значения. Если входной тензор не является
комплексным, функция возвращает его копию без изменений.
Синтаксис
torch.real(input)
где input - комплексный тензор.
Пример
Давайте создадим комплексный тензор и извлечем его действительную часть:
import torch
t = torch.tensor([1 + 2j, 3 + 4j, 5 + 6j])
res = torch.real(t)
print(res)
Результат выполнения кода:
tensor([1., 3., 5.])
Пример
Функция работает и с двумерными комплексными тензорами:
import torch
t = torch.tensor([
[1 + 2j, 3 + 4j],
[5 + 6j, 7 + 8j]
])
res = torch.real(t)
print(res)
Результат выполнения кода:
tensor([
[1., 3.],
[5., 7.]
])
Пример
Если передать в функцию действительный тензор, она вернет его копию:
import torch
t = torch.tensor([1.0, 2.0, 3.0])
res = torch.real(t)
print(res)
Результат выполнения кода:
tensor([1., 2., 3.])
