torch.fft.ifftshift¶

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

fftshift() 的逆运算。

参数

input (Tensor) – 以 FFT 顺序排列的张量
dim (int, Tuple[int], 可选) – 要重新排列的维度。仅重新排列此处指定的维度，任何其他维度都将保留其原始顺序。默认值：input 的所有维度。

示例

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

通过 fftshift() 和 ifftshift() 的往返过程得到相同的结果

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

文档