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])