Mam macierz n-m-m, którą chcę przekonwertować na macierz mn-m-m, przy czym każdy blok m-m wyniku zawiera przekątną każdego wiersza.

Na przykład, jeśli dane wejściowe to:

[1 2; 3 4; 5 6]

wyjście powinno być:

[1 0; 0 2; 3 0; 0 4; 5 0; 0 6]

Oczywiście nie chcę składać matrycy krok po kroku za pomocąfor pętla.
Czy istnieje wektoryzowany i prosty sposób na osiągnięcie tego celu?