Python torch.fft.rfft()函数用法示例代码
作者:Oxygen?H2O 时间:2022-02-15 02:03:36
在新旧版的torch中的傅里叶变换函数在定义和用法上存在不同,记录一下。
1、旧版
fft = torch.rfft(input, 2, normalized=True, onesided=False)
# input 为输入的图片或者向量,dtype=torch.float32,size比如为[1,3,64,64]
# signal_ndim(int):The number of dimensions in each signal,can only be 1、2、3
# normalized(bool,optional):controls wheather to return normallized results. Default:False
# onesided(bool,optional):controls whether to return half of results to avoid redundancy.Default:True
上面例子中图像中 singal_ndim = 2 ,是因为输入图像是2维的。
1.7之后的版本中,如果要用 oneside output,则改用torch.fft.rfft();如果要用two-side output,则改用torch.fft.fft()
input= torch.arange(4)
fft = torch.rfft(input, 2, normalized=True, onesided=False)
2、新版
一维离散傅里叶变换
torch.fft.rfft(input,n=None,dim=-1,norm=None) --> Tensor
# input:Tensor
# n(int,optional):Output signal length. This determines the length of the
output signal.
# dim(int, optional): The dimension along which to take the one dimensional real IFFT.
# norm (str, optional): Normalization mode.
二维离散傅里叶变换
torch.fft.rfft2(input, s=None, dim=(-2, -1), norm=None, *, out=None) -> Tensor
input (Tensor): the input tensor
s (Tuple[int], optional): Signal size in the transformed dimensions.
dim (Tuple[int], optional): Dimensions to be transformed.
norm (str, optional): Normalization mode.
高维离散傅里叶变换
rfftn(input, s=None, dim=None, norm=None, *, out=None) -> Tensor
input (Tensor): the input tensor
s (Tuple[int], optional): Signal size in the transformed dimensions.
dim (Tuple[int], optional): Dimensions to be transformed.
norm (str, optional): Normalization mode. For the forward transform
3、新旧版对比
import torch
input = torch.rand(1,3,32,32)
# 旧版pytorch.rfft()函数
fft = torch.rfft(input, 2, normalized=True, onesided=False)
# 新版 pytorch.fft.rfft2()函数
output = torch.fft.fft2(input, dim=(-2, -1))
output = torch.stack((output.real, output_new.imag), -1)
ffted = torch.rfft(input, 1, onesided=False) to ffted = torch.view_as_real(torch.fft.fft(input, dim=1))
and
iffted = torch.irfft(time_step_as_inner, 1, onesided=False) to
iffted = torch.fft.irfft(torch.view_as_complex(time_step_as_inner), n=time_step_as_inner.shape[1], dim=1)
补充:使用numpy模拟torch.fft.fft拯救paddle
import numpy as np
import torch
import paddle
def paddle_fft(x,dim=-1):
if dim==-1:
return paddle.to_tensor(np.fft.fft(x.numpy()))
else:
shape= [i for i in range(len(x.shape))]
shape[dim],shape[-1]=shape[-1],shape[dim]
x=np.transpose(np.fft.fft(np.transpose(x.numpy(), shape)),shape)
return paddle.to_tensor(x)
if __name__ == '__main__':
data=paddle.to_tensor(np.array([[[1, 4, 3], [1, 2, 3]], [[1, 2, 3], [1, 2, 3]]]))
paddle_f_d=paddle_fft(paddle_fft(data,-1),-2)
torch_f_d =paddle_fft(torch.fft.fft(torch.Tensor(data.numpy()),dim=-1),-2)
print(paddle_f_d.numpy())
print(torch_f_d.numpy())
来源:https://blog.csdn.net/oxygenh2o/article/details/122157814
标签:torch.fft.rfft(),函数,python
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
html中的sub与sup标签
2009-03-06 13:12:00
![](https://img.aspxhome.com/file/UploadPic/20093/6/120757af09ag213-77s.jpg)
深入浅出的SQL server 查询优化
2010-07-02 20:58:00
Javascript语法检查插件 jsLint for Vim
2009-03-11 16:37:00
![](https://img.aspxhome.com/file/UploadPic/20093/11/medium-86s.jpg)
ubuntu 16.04下python版本切换的方法
2021-07-10 16:36:04
![](https://img.aspxhome.com/file/2023/1/67871_0s.png)
ASP中如何判断一个字符是不是汉字
2008-05-04 12:47:00
IE6,7下实现white-space:pre-wrap;
2009-12-31 18:30:00
![](https://img.aspxhome.com/file/UploadPic/20101/11/01-79s.png)
脚本安全的本质_PHP+MYSQL第1/3页
2023-11-23 23:54:45
详解tensorflow载入数据的三种方式
2023-07-22 19:35:56
![](https://img.aspxhome.com/file/2023/4/62904_0s.jpg)
解析Extjs与php数据交互(增删查改)
2023-07-19 09:06:42
使用SQL Server 2008中对象相关性
2008-12-12 06:36:00
10个杀手级应用的Python自动化脚本
2023-10-06 14:01:53
Python Spyder 调出缩进对齐线的操作
2023-07-17 00:56:04
![](https://img.aspxhome.com/file/2023/2/61172_0s.jpg)
PHP生成静态页面详解
2023-11-21 06:50:43
python实现简单成绩录入系统
2023-09-16 08:00:27
PHP登录验证功能示例【用户名、密码、验证码、数据库、已登陆验证、自动登录和注销登录等】
2023-11-14 23:24:02
![](https://img.aspxhome.com/file/2023/5/86925_0s.png)
asp读取xml实例代码
2011-03-08 11:13:00
django初始化数据库的实例
2023-09-19 07:35:47
安装PHP遇到“无法载入mysql扩展”解决方法
2007-06-15 15:04:00
pycharm必知的一些简单设置方法
2023-07-21 14:53:40
![](https://img.aspxhome.com/file/2023/7/61967_0s.png)
简单介绍Python中的struct模块
2023-09-22 16:27:01