Matlab:加速大for循环

Question

我正在尝试将大型矩阵中的 3 x 3 掩码转换为向量。目前，这是我的代码:

A=rand(3264,2448)

[rows cols]=size(A);

T=zeros(9,prod(size(A)-2));

for i=1:rows-2

  for j=1:cols-2

    T(:,(i-1)*cols+j)=reshape(A(i:i+2,j:j+2),[9 1]);

  end

end

目前，这需要相当长的时间。我能以任何方式加快速度吗？我在考虑 meshgrid 和 arrayfun，但无法完全理解它们。

谢谢!

Answer 1

您可以使用 im2col将图像 block 重新排列成列，例如:

T =  im2col(A,[3 3],'sliding');