matlab gpu for循环_MATLAB中三维矩阵相乘的阳间写法和阴间写法速度对比

本文探讨了在MATLAB中进行三维矩阵相乘时,使用for循环、arrayfun以及深度学习工具箱的pagemtimes函数的效率差异。结果显示,pagemtimes在2020b版本中速度最快,且GPU计算(pagefun)优于CPU。建议在处理此类问题时考虑升级MATLAB版本和利用GPU计算。
摘要由CSDN通过智能技术生成

819da013b3c1f6e47561c3b630f36f54.png

之前写 Clark 变换和 Park 变换的时候,写过一个三维矩阵的相乘,为了不写循环绕了一个非常大的弯,现在试试正常写循环,用 arrayfun 省循环,用自动扩展乘法并加和省循环,三种速度都是个啥水平。

直接上代码

clear

matsize = 50000;
a = rand(100,100,matsize);
b = rand(100,1,matsize);

tic
c = zeros(100,1,matsize);
for i = 1:matsize
    c(:,:,i) = a(:,:,i)*b(:,:,i);
end
c = squeeze(c);
toc

tic
d = cell2mat(arrayfun(@(i)a(:,:,i)*b(:,:,i),1:matsize,'UniformOutput',false));
toc

tic
e = squeeze(sum(a.*permute(b,[2,1,3]),2));
toc

运行结果如下

>> test_speed
历时 0.665446 秒。
历时 0.862202 秒。
历时 0.881269 秒。

不愧是阴间写法,速度最慢,最难理解。

<
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值