Python PyTorch ifftshift用法及代碼示例

本文簡要介紹python語言中 torch.fft.ifftshift 的用法。

用法:

torch.fft.ifftshift(input, dim=None) → Tensor

參數：

• input(Tensor) -FFT 順序的張量

• dim(int,元組[int],可選的) - 要重新排列的尺寸。隻有此處指定的尺寸會重新排列，任何其他尺寸都將保留其原始順序。默認值：所有維度input.

fftshift() 的倒數。

示例

>>> f = torch.fft.fftfreq(5)
>>> f
tensor([ 0.0000,  0.2000,  0.4000, -0.4000, -0.2000])

round-trip 到 fftshift() 和 ifftshift() 給出相同的結果：

>>> shifted = torch.fft.fftshift(f)
>>> torch.fft.ifftshift(shifted)
tensor([ 0.0000,  0.2000,  0.4000, -0.4000, -0.2000])