当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python dask.array.fft.rfftn用法及代码示例


用法:

dask.array.fft.rfftn(a, s=None, axes=None)

numpy.fft.rfftn 的包装

应用 FFT 的轴必须只有一个块。要更改数组的分块,请使用 dask.Array.rechunk。

numpy.fft.rfftn 文档字符串如下:

计算实际输入的 N 维离散傅里叶变换。

此函数通过快速傅里叶变换 (FFT) 在 M-dimensional 实数数组中的任意数量的轴上计算 N 维离散傅里叶变换。默认情况下,所有轴都被变换,实际变换在最后一个轴上执行,而其余变换是复杂的。

参数

aarray_like

输入数组,被认为是真实的。

s整数序列,可选

要从输入中使用的形状(沿每个变换轴的长度)。 (s[0] 指轴 0,s[1] 指轴 1,等等)。 s 的最后一个元素对应于 rfft(x, n)n ,而对于其余轴,它对应于 fft(x, n)n 。沿着任何轴,如果给定的形状小于输入的形状,则裁剪输入。如果它更大,则用零填充输入。如果未给出s,则使用沿axes 指定的轴的输入形状。

axes整数序列,可选

计算 FFT 的轴。如果未指定,则使用最后的 len(s) 轴,如果也未指定 s,则使用所有轴。

norm{“backward”, “ortho”, “forward”},可选

规范化模式(参见numpy.fft)。默认为“backward”。指示前向/后向变换对的哪个方向被缩放以及使用什么归一化因子。

返回

out复杂的ndarray

截断或补零的输入,沿 axessa 的组合指示的轴进行转换,如上面的参数部分所述。最后一个变换轴的长度将为 s[-1]//2+1 ,而其余变换轴的长度将根据 s ,或与输入保持不变。

抛出

ValueError

如果saxes 的长度不同。

索引错误

如果 axes 的元素大于 a 的轴数。

注意

实际输入的变换在最后一个变换轴上执行,如 rfft ,然后在其余轴上的变换如 fftn 。输出的顺序与 rfft 用于最终转换轴,与 fftn 用于剩余转换轴。

有关使用的详细信息、定义和约定,请参阅fft

例子

>>> a = np.ones((2, 2, 2))  
>>> np.fft.rfftn(a)  
array([[[8.+0.j,  0.+0.j], # may vary
        [0.+0.j,  0.+0.j]],
       [[0.+0.j,  0.+0.j],
        [0.+0.j,  0.+0.j]]])
>>> np.fft.rfftn(a, axes=(2, 0))  
array([[[4.+0.j,  0.+0.j], # may vary
        [4.+0.j,  0.+0.j]],
       [[0.+0.j,  0.+0.j],
        [0.+0.j,  0.+0.j]]])

相关用法


注:本文由纯净天空筛选整理自dask.org大神的英文原创作品 dask.array.fft.rfftn。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。