真不好意思,才隔一天就又上来问..
昨天问的东西其实是我要算的东西的第二步,原本他才是瓶颈,但是板上高手一下就解决
了,我现在想不用for处理第一步,也就是求出昨天的atot (m*n,k)
atot中的元素是用一个m*n二维座标(x,y)和三个一组的数字(a1,a2,a3)带入一个函数
f(x,y,a1,a2,a3)求得,三个一组的数字共有k组,所以最后atot的维度是(m*n,k)
https://imgur.com/CTjyecs
程式上先用meshgrid建立座标,再产生k组数字
m=15;
n=19;
k=10000;
[X,Y]=meshgrid(1:m,1:n);
ini=randn(k,3);
为了方便起见这里f就先用sin和多项式来举例
f(x,y,a1,a2,a3)=sin(x+y^2+a1+a2^2+a3^3)
目前想了一个办法是用arrayfun配上reshape
atotpr=arrayfun(@(x,y) sin(x+y^2+ini(:,1)+ini(:,2).^2+ini(:,3).^3),X,Y,'un',0);
atotmid=cell2mat(atotpr);
atot=reshape(atotmid,k,m*n);
这样确实比用三层for loop快很多了,随便试几个m,n,k都快一个数量级以上,不过想知道
有没有更好的方法。另外这样写gpu也没有办法算,matlab的gpu不接受non UniformOutput
,虽然说以目前的速度来看,gpu似乎也不是很有必要了....但还是希望能多学一点
感恩