我正在尝试沿大矩阵（例如 10x10）的对角线对小矩阵（2x2）进行元素插入。添加重叠值，并且仅将小矩阵插入到可以完全放入大矩阵的位置。

我已经使用 for 循环实现了这一点，但我很好奇该过程是否可以矢量化。

function M = TestDiagonal()

N     = 10;
miniM = [1, -1; -1, 1];
M     = zeros(N);

for i = 1:N-1
    M(i:i+1,i:i+1) = M(i:i+1,i:i+1) + miniM;
end

end

给出所需的矩阵

 1    -1     0     0     0     0     0     0     0     0
-1     2    -1     0     0     0     0     0     0     0
 0    -1     2    -1     0     0     0     0     0     0
 0     0    -1     2    -1     0     0     0     0     0
 0     0     0    -1     2    -1     0     0     0     0
 0     0     0     0    -1     2    -1     0     0     0
 0     0     0     0     0    -1     2    -1     0     0
 0     0     0     0     0     0    -1     2    -1     0
 0     0     0     0     0     0     0    -1     2    -1
 0     0     0     0     0     0     0     0    -1     1

在一般情况下，输入始终是正方形，但可以采用任何大小。步长维度始终等于 1。

只需使用 2D 卷积（参见conv2 https://www.mathworks.com/help/matlab/ref/conv2.html).

2×2 情况，沿每个维度步骤 1

M = conv2(eye(N-1), miniM);

m×m案例，沿每个维度的步骤 1

M = conv2(eye(N-size(miniM-1)+1), miniM);

m×n情况下，沿每个维度的任意步骤

在这种情况下，需要定义步骤：

step = [2 1]; % desired step along each dimension

定义所需的重复次数更有意义R，而不是最终尺寸（N），因为后者可能无法通过完全重复来实现miniM:

R = 4; % desired number of repetitions

Then:

M = conv2(full(sparse(1:step(1):step(1)*R, 1:step(2):step(2)*R, 1)), miniM);

Example:

>> miniM = [10 20 30; 40 50 60];
>> R = 4;
>> step = [1 2];
>> M = conv2(full(sparse(1:step(1):step(1)*R, 1:step(2):step(2)*R, 1)), miniM)
M =
    10    20    30     0     0     0     0     0     0
    40    50    70    20    30     0     0     0     0
     0     0    40    50    70    20    30     0     0
     0     0     0     0    40    50    70    20    30
     0     0     0     0     0     0    40    50    60