Matlab：加速大循环

Question

我正试图从大矩阵中将3×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将图像块重新排列为列，例如：

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