Fast Matrix-Vector-Multiplication By FFT

自我o救赎

已于 2023-03-03 10:55:21 修改

阅读量122

点赞数

分类专栏： Python语言专栏文章标签： python

于 2022-11-15 10:26:46 首次发布

本文链接：https://blog.csdn.net/y15520833229/article/details/127860969

版权

Python语言专栏专栏收录该内容

69 篇文章 2 订阅

订阅专栏

在这里插入图片描述

>> l = rand(1000,1);
>> l1 = l';
>> C = toeplitz([l1(1) fliplr(l1(2:end))], l1);
>> C(1:5, 1:5)

ans =

    0.8147    0.9058    0.1270    0.9134    0.6324
    0.8667    0.8147    0.9058    0.1270    0.9134
    0.0600    0.8667    0.8147    0.9058    0.1270
    0.4440    0.0600    0.8667    0.8147    0.9058
    0.9509    0.4440    0.0600    0.8667    0.8147
>> x = rand(1000,1);
>> y=C *x;
>> for i = 1:100, y=C *x; end;
>> tic; y=C *x; toc;
Elapsed time is 0.000913 seconds.
>> tic; for i = 1:100, y=C *x; end; toc
Elapsed time is 0.074619 seconds.

在这里插入图片描述

>> y1 = ifft(fft(C(1:end,1)) .* fft(x));
>> norm(y-y1)

ans =

   4.4646e-12
>> ll = C(1:end,1);
>> tic; for i = 1:100, y1 = ifft(fft(ll) .* fft(x)); end; toc
Elapsed time is 0.011608 seconds.
误差非常小，同时运算速度得到了几倍的提升，大约 0.1 ms/loop

在这里插入图片描述

>> l = rand(1000,1);
>> T = toeplitz(l, [l(1); zeros(999,1)]);
>> T(1:5, 1:5)

ans =

    0.8147         0         0         0         0
    0.9058    0.8147         0         0         0
    0.1270    0.9058    0.8147         0         0
    0.9134    0.1270    0.9058    0.8147         0
    0.6324    0.9134    0.1270    0.9058    0.8147
>> x = rand(1000,1);
>> tic; y = T*x; toc
Elapsed time is 0.000593 seconds.
>> tic; for i = 1:100, y = T*x; end; toc
Elapsed time is 0.010883 seconds.

当然我们还可以把矩阵变成稀疏矩阵

>> T1 = sparse(T);
>> tic; y = T1*x; toc
Elapsed time is 0.826337 seconds.
>> tic; for i = 1:100, y = T1*x; end; toc
Elapsed time is 0.033503 seconds.
>> tic; y = T1*x; toc
Elapsed time is 0.000596 seconds.

在这里插入图片描述

>> y1 = T*x;
>> y2 = ifft(fft(l1).*fft([x;zeros(1000,1)]));
>> norm(y2(1:1000) - y1)

ans =

   3.4828e-12
>> tic; y2 = ifft(fft(l1).*fft([x;zeros(1000,1)])); toc
Elapsed time is 0.000527 seconds.
>> tic; for i = 1:100,  y2 = ifft(fft(l1).*fft([x;zeros(1000,1)])); end; toc
Elapsed time is 0.005841 seconds.