我有一个双循环,这是非常低效的. c is a [400,2000] matrixr is a [2000,1] matrixP is a [2000,1] matrixS is a [1, 400] matrixfor i=1:400 for k=1:2000 c(i,k) = r(k,1) * max([0, P(k,1) - S(1,i)]); endend 我试图制作一个parfor并
c is a [400,2000] matrix
r is a [2000,1] matrix
P is a [2000,1] matrix
S is a [1, 400] matrix
for i=1:400
for k=1:2000
c(i,k) = r(k,1) * max([0, P(k,1) - S(1,i)]);
end
end
我试图制作一个parfor并且它有效.但我一直在寻找更优雅的解决方案.我一直在努力尝试但没有运气……
由于您只进行元素操作,如 – 和.*,这需要使用bsxfun的解决方案.
使用
bsxfun(@minus,P,S)
做元素减法P(k,1) – S(1,i).输出将是[2000,400]矩阵.您可以对此矩阵应用max(0,…)运算,最后再次使用bsxfun将每一行乘以相应的r：
bsxfun(@times,max(bsxfun(@minus,P,S),0),r)
由于你的c应该是[400,2000]的大小,添加一个最终的转置操作,你就完成了.
c = bsxfun(@times,max(bsxfun(@minus,P,S),0),r).';
一个小的时间比较：for循环需要
Elapsed time is 0.688408 seconds.
而bsxfun解决方案只需要
Elapsed time is 0.007884 seconds.
对于完全相同的结果,这是一个很好的加速87.