test_fft.py 文件源码

python
阅读 37 收藏 0 点赞 0 评论 0

项目:pyculib 作者: numba 项目源码 文件源码
def test_fft_2d_single_col_major(self):
        from pyculib.fft import fft
        N2 = 2
        N1 = 8
        N = N1 * N2
        x = np.asarray(np.arange(N), dtype=np.float32).reshape((N2, N1), order='F')
        xf_ref = np.fft.rfft2(x)

        xf = np.empty(shape=(N2, N1//2 + 1), dtype=np.complex64, order='F')
        fft(x, xf)
        self.assertTrue( np.allclose(xf_ref, xf, atol=1e-6) )
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号