在numpy中沿一个轴对角化二维矩阵

【问题标题】：Diagonalize 2d matrix along one axis in numpy在numpy中沿一个轴对角化二维矩阵
【发布时间】：2017-06-14 10:03:00
【问题描述】：

给定矩阵 X，维度为 MxN。我想为 X 的每一行创建一个对角矩阵。结果应该是 MxNxN。如何有效地做到这一点？谢谢！

【问题讨论】：

【解决方案1】：

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

【讨论】：