Diagonalize 2d Matrix entlang einer Achse in numpy

Question

Gegeben die Matrix X in der Dimension von MxN. Ich möchte für jede Zeile von X eine diagonale Matrix erstellen. Das Ergebnis sollte in MxNxN sein. Wie man es effizient macht? Vielen Dank!

Answer 1

out = np.zeros((m, n, n)) 
out[:, np.arange(n), np.arange(n)] = X