Как сделать блочную разреженную матрицу из n блоков размера?

Question

Существует ли быстрый способ заполнить (разреженную) матрицу блоками A и B как таковые в Matlab:

 A, B, 0, 0, 0
 B, A, B, 0, 0
 0, B, A, B, 0
 0, 0, B, A, B
 0, 0, 0, B, A

Я пробовал этот код

e=full([A B; B A]);
e1=e;
for i=1:n
   e=blkdiag(e,e1);
end 

Answer 1

Вы можете сделать следующее (сохранить функцию в файл blktritoep.m):

function C = blktritoep(A, B, nb)
% A, B must be square of and same size for this to work
% nb = number of times A gets repeated on the diagonal of C
A = sparse(A); % Does nothing if A is already sparse
B = sparse(B); % Idem
C = kron(diag(ones(nb,1)), A) + kron(diag(ones(nb-1,1), 1), B) + ...
    kron(diag(ones(nb-1,1), -1), B);
end

Вывод, который я получаю, скажем,

A = ones(2,2);
B = -2 * ones(2,2);
C = blktritoep(A, B, 3);
full(C) % Result is sparse, just for pretty printing!

есть

ans =

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