torch.fft.ihfft2

torch.fft.ihfft2(input, s=None, dim=(-2, -1), norm=None, *, out=None) → Tensor

Вычисляет двумерное обратное дискретное преобразование Фурье вещественного input. Эквивалентно ihfftn(), но по умолчанию преобразует только две последние размерности.

Примечание

Поддерживает torch.half на CUDA с архитектурой GPU SM53 или выше. Однако он поддерживает только длину сигнала, являющуюся степенью двойки, во всех преобразуемых размерностях.

Параметры

• input (Tensor) – входной тензор
• s (Tuple[int], optional) – Размер сигнала в преобразуемых размерностях. Если задано, каждая размерность dim[i] будет либо дополнена нулями, либо обрезана до длины s[i] перед вычислением обратного преобразования Гермита. Если задана длина -1 , в этой размерности дополнение нулями не выполняется. Значение по умолчанию: s = [input.size(d) for d in dim]
• dim (Tuple[int], optional) – Размерности, которые необходимо преобразовать. По умолчанию: две последние размерности.

• norm (str, optional) –

  Режим нормализации. Для обратного преобразования (ihfft2()) эти соответствуют:

  • "forward" - без нормализации
  • "backward" - нормализация по 1/n
  • "ortho" - нормализация по 1/sqrt(n) (делая обратное преобразование Гермита ортогональным)

  Где n = prod(s) - логический размер обратного преобразования. Вызов прямого преобразования (hfft2()) со всем тем же режимом нормализации применит общую нормализацию 1/n между двумя преобразованиями. Это необходимо для того, чтобы ihfft2() было точной обратной операцией.

  Значение по умолчанию "backward" (нормализация по 1/n).

Ключевые аргументы

out (Tensor, optional) – выходной тензор.

Пример

>>> T = torch.rand(10, 10)
>>> t = torch.fft.ihfft2(t)
>>> t.size()
torch.Size([10, 6])

В сравнении с полным выходом из ifft2(), сигнал Гермита в временном пространстве занимает только половину места.

>>> fftn = torch.fft.ifft2(t)
>>> torch.allclose(fftn[..., :6], rfftn)
True

Дискретное преобразование Фурье является разделяемым, поэтому ihfft2() здесь эквивалентно комбинации ifft() и ihfft():

>>> two_ffts = torch.fft.ifft(torch.fft.ihfft(t, dim=1), dim=0)
>>> torch.allclose(t, two_ffts)
True