2017-01-28 12 views
2

MxNの次元が行列Xであると仮定すると、私はXのすべての行に対して対角行列を作成したいと思います。結果はMxNxNでなければなりません。どのように効率的にそれを行うには?ありがとうございました!numpyで1軸に沿って2d行列を対角化する

答えて

3
out = np.zeros((m, n, n)) 
out[:, np.arange(n), np.arange(n)] = X 
関連する問題