Matlab 如何在存在for循环和矩阵向量乘法的情况下减少计算时间

我正在写一个程序，其中计算时间非常重要，所以我必须以一种减少时间的方式编写代码。在下面，我写了一段代码，但是如果向量的长度变长，这将非常耗时。有没有办法以更快的方式产生同样的结果

K1 = [1 2 3 4 5];  K2 = [6 7 8 9 10];
kt1 = [1.5 3 4.5]; kt2 = [6.5 8 9.5];
numk1 = bsxfun(@minus,K1.',kt1);
denomk1 = bsxfun(@minus, kt1.',kt1)+eye(numel(kt1));
numk2 = bsxfun(@minus,K2.',kt2);
denomk2 = bsxfun(@minus, kt2.', kt2)+eye(numel(kt2));

for j=1:numel(kt1)
    for jj=1:numel(kt2)
        k1_dir = bsxfun(@rdivide,numk1,denomk1(j,:)); k1_dir(:,j)=[];
        k_dir1 = prod(k1_dir,2);
        k2_dir = bsxfun(@rdivide,numk2,denomk2(jj,:)); k2_dir(:,jj)=[];
        k_dir2 = prod(k2_dir,2);
        k1_k2(:,:,j,jj) = k_dir1 * k_dir2';
    end
end 

在上述代码中，随着

K1

和

K2

的长度增加，

kt1

和

kt2

的长度也增加。因此，对于较长的向量长度，此代码非常耗时

开足马力，这可能是一种替代问题中列出的代码的循环部分的方法-

%// Size parameters
M1 = numel(K1);
N1 = numel(kt1);
M2 = numel(K2);
N2 = numel(kt2);

%// Indices to be removed from k1_dir & k2_dir. 
%// In our vectorized version, we will just use these to set 
%// corresponding elements in vectorized versions of k1_dir & k2_dir
%// to ONES, as later on PROD would take care of it.
rm_idx1 = bsxfun(@plus,[1:M1]',[0:N1-1]*(M1*N1+M1)); %//'
rm_idx2 = bsxfun(@plus,[1:M2]',[0:N2-1]*(M2*N2+M2)); %//'

%// Get vectorized version of k1_dir, as k1_dirv
k1_dirv = bsxfun(@rdivide,numk1,permute(denomk1,[3 2 1]));
k1_dirv(rm_idx1) = 1;
k_dir1v = prod(k1_dirv,2);

%// Get vectorized version of k2_dir, as k2_dirv
k2_dirv = bsxfun(@rdivide,numk2,permute(denomk2,[3 2 1]))
k2_dirv(rm_idx2) = 1;
k_dir2v = prod(k2_dirv,2);

%// Get vectorized version of k1_k2, as k1_k2v
k1_k2v = bsxfun(@times,k_dir1v,permute(k_dir2v,[2 1 4 3]));

---

快速运行时测试：

输入设置如下-

SZ1 = 100;
SZ2 = 100;

K1 = randi(9,1,SZ1);
K2 = randi(9,1,SZ1);

kt1 = randi(9,1,SZ2);
kt2 = randi(9,1,SZ2);

原始版本中循环部分的运行时（在添加代码以进行更公平的基准测试时使用零进行预分配之后）和建议的矢量化方法如下所示-

-------------------------- With Original Loopy Approach
Elapsed time is 1.086666 seconds.
-------------------------- With Proposed Vectorized Approach
Elapsed time is 0.178805 seconds.

---

似乎JIT并没有显示出它的魔力，至少在嵌套循环中使用bsxfun时是这样，而且在每次迭代中都需要索引到那个巨大的4D数组中这一事实对您没有帮助。所以，在这种情况下，全速进行矢量化更有意义

当

K1

和

K2

的维度不同时，是否可以修改它以使其工作？

permute

在4个维度中：O无注释，+1:）@santansalai怎么样？：）@SanthanSalai我使用的

permute

解决方案可以通过找到。看起来你已经有了答案，但在将来遇到性能问题时，请具体说明。”“大的时候慢”不是很相关，“我的输入大小为9,1100时1秒”帮助人们知道你需要什么。@DennisJaheruddin谢谢你的评论，我会这样做的