2
MxN의 차원에서 행렬 X가 주어 졌을 때. X의 모든 행에 대해 대각선 행렬을 만들고 싶습니다. 결과는 MxNxN이어야합니다. 그것을 효율적으로하는 방법? 고맙습니다!numpy에서 한 축을 따라 2 차원 행렬을 대각 화합니다.
A
답변
3
out = np.zeros((m, n, n))
out[:, np.arange(n), np.arange(n)] = X