torch.fft.fft2

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

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

Примечание

Представление в частотной области любого вещественного сигнала удовлетворяет свойству эрмитовости: X[i, j] = conj(X[-i, -j]). Эта функция всегда возвращает все положительные и отрицательные частотные компоненты, хотя для вещественных входных данных половина этих значений избыточна. rfft2() возвращает более компактное одностороннее представление, в котором возвращаются только положительные частоты последнего измерения.

Примечание

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

Параметры

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

• norm (строка, необязательно) –

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

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

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

  По умолчанию "backward" (без нормализации).

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

out (Тензор, необязательно) – выходной тензор.

Пример

>>> x = torch.rand(10, 10, dtype=torch.complex64)
>>> fft2 = torch.fft.fft2(x)

Дискретное преобразование Фурье является разложимым, поэтому fft2() здесь эквивалентно двум одномерным вызовам fft():

>>> two_ffts = torch.fft.fft(torch.fft.fft(x, dim=0), dim=1)
>>> torch.testing.assert_close(fft2, two_ffts, check_stride=False)