在numpy中沿着一個軸對角化2d矩陣

給定矩陣X的維數爲MxN。我想爲X的每一行創建一個對角矩陣，結果應該在MxNxN中。如何有效地做到這一點？謝謝！在numpy中沿着一個軸對角化2d矩陣

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

2017-01-28 22:52:24

回答