比cell2mat更快

时间:2015-11-03 08:26:23

标签: matlab cell

我目前正在优化某些代码,并且无法找到比MATLAB cell2mat更快的方法。目前,我的代码cell2mat的多次使用占处理时间的15%以上

我认为它可以比这更快,因为我知道单元数组的结构我将传递给函数。

基本上,单元格数组为NxN,其中:

  1. 左上角(N-1)x(N-1)块在每个单元格中包含6x6双矩阵

  2. 右下角(N,N)单元格是MxM双矩阵。

  3. 其他单元格具有正确的连接尺寸,即:

  4. 单元格(1:(N-1),N)6xM双矩阵,单元格(N,1:(N-1))Mx6双矩阵。 (为了清晰起见添加了图片,N=207M=300

    enter image description here

    由于单元格将始终填充双精度并且始终为2维,因此我只使用了cell2mat代码的一小部分,即:

     function m = myCell2Mat(c)
    
        rows = size(c,1);
    
    
            m = cell(rows,1);
            % Concatenate one dim first
            for n=1:rows
                m{n} = cat(2,c{n,:});% 73% of the time spent here
            end
            % Now concatenate the single column of cells into a matrix
            m = cat(1,m{:});% 25.2% of the time spent there
    
    
      end
    

    这样做会改变花费的时间(就像人们想象的那样),花在这些线上的时间最多。

    我的问题是:有没有人知道如何删除那里的循环?我尝试了以下几点:

    N=207;
    M=300;
    C=cell(N,N);
    for ii=1:N-1
      for jj=1:N-1
    C{ii,jj}=rand(6);
      end
    end
    
    for kk=1:(N-1)
    C{N,kk}=rand(M,6);
    C{kk,N}=rand(6,M);
    end
    
    C{end}=rand(M,M);
    
    tmp1=cat(1,C{:,1:(end-1)});
    LeftPart=reshape(tmp1,[],6*(size(C,2)-1));
    
    RightPart=cat(1,C{:,end});
    
    Res=[LeftPart RightPart];
    

    但它并没有及时显示任何改进。(并且reshape按列操作时会给出错误的结果)

    我已经考虑过使用递归函数,但它似乎没有去过任何地方。

    提前致谢!

1 个答案:

答案 0 :(得分:4)

评论的结论是,没有办法显着改善cell2mat的表现。相反,我会建议一个更好的数据结构。

您写道,数据实际上代表了2D-Matrix。不是将其拆分成块,而是将其重新整形为4D矩阵的速度更快。我们假设你的Cell被称为CM是相应的2D矩阵

%Old Code
q=C{1,1}
%Faster way to index the same
%1) Convert 2D matrix to 4D
blocksize=6
M=reshape(M,blocksize,size(M,1)/blocksize,blocksize,size(M,2)/blocksize);
%2) Index a block
q=squeeze(M(:,1,:,1))

由于重塑基本上没有时间,您可以切换回2D视图,例如处理最后一行/列。如果您想避免转换回来,可以一次索引多个块。对于最后一列,请使用:reshape(M(:,206,:,207:end),6,[]),对应C{206,207}

相关问题