pytorch 笔记：torch.fft-Toy模板网

这篇具有很好参考价值的文章主要介绍了pytorch 笔记：torch.fft。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

1 FFT

torch.fft,pytorch学习,pytorch,深度学习,人工智能

进行一个维度的快速傅里叶变换

torch.fft.fft(input, 
                n=None, 
                dim=- 1, 
                norm=None, 
                *, 
                out=None)

1.1 主要参数

input	输入，需要傅里叶变换的tensor
n	需要变换的tensor的长度，默认是input的长度如果比input长度大，那么补0 如果比input长度小，那么截取
dim	哪一个维度进行快速傅里叶变换
norm	‘forward’：用1/n 正则化 ‘backward’：无需正则化 ‘ortho’：使用1/sqrt(n)正则化

1.2 举例

import torch

a=torch.Tensor([0,1,2,3,4])
torch.fft.fft(a)
'''
tensor([10.0000+0.0000j, -2.5000+3.4410j, -2.5000+0.8123j, -2.5000-0.8123j,
        -2.5000-3.4410j])
'''

2 iFFT

逆正则化

torch.fft.ifft(
    input, 
    n=None, 
    dim=- 1, 
    norm=None, 
    *, 
    out=None)

2.1 主要参数

input	输入，需要傅里叶变换的tensor
n	需要变换的tensor的长度，默认是input的长度如果比input长度大，那么补0 如果比input长度小，那么截取
dim	哪一个维度进行快速傅里叶变换
norm	‘forward’：用1/n 正则化 ‘backward’：无需正则化 ‘ortho’：使用1/sqrt(n)正则化

2.2 举例

import torch

a=torch.Tensor([[0,1,2,3],
                [3,4,6,7]])
torch.fft.ifft(torch.fft.fft(a))
'''
tensor([[0.+0.j, 1.+0.j, 2.+0.j, 3.+0.j],
        [3.+0.j, 4.+0.j, 6.+0.j, 7.+0.j]])
'''

3 rFFt

将FFT结果中虚数部分为负，且有对应共轭的那一部分去除，减少存储量

torch.fft.rfft(
    input, 
    n=None, 
    dim=- 1, 
    norm=None, 
    *, 
    out=None)

3.1 主要参数

input	输入，需要傅里叶变换的tensor
n	需要变换的tensor的长度，默认是input的长度如果比input长度大，那么补0 如果比input长度小，那么截取
dim	哪一个维度进行快速傅里叶变换
norm	‘forward’：用1/n 正则化 ‘backward’：无需正则化 ‘ortho’：使用1/sqrt(n)正则化

3.2 举例（和FFT的对比）

import torch

a=torch.Tensor([[0,1,2,3],
                [3,4,6,7]])
torch.fft.fft(a)
'''
tensor([[ 6.+0.j, -2.+2.j, -2.+0.j, -2.-2.j],
        [20.+0.j, -3.+3.j, -2.+0.j, -3.-3.j]])
'''

torch.fft.rfft(a)
'''
tensor([[ 6.+0.j, -2.+2.j, -2.+0.j],
        [20.+0.j, -3.+3.j, -2.+0.j]])
'''

4 irfft

rfft的逆运算文章来源地址https://www.toymoban.com/news/detail-756973.html

import torch

a=torch.Tensor([[0,1,2,3],
                [3,4,6,7]])
torch.fft.irfft(torch.fft.rfft(a))
'''
tensor([[0., 1., 2., 3.],
        [3., 4., 6., 7.]])
'''

到了这里，关于pytorch 笔记：torch.fft的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！