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