沿numpy沿一个轴对角化2d矩阵 [英] Diagonalize 2d matrix along one axis in numpy
本文介绍了沿numpy沿一个轴对角化2d矩阵的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
给出矩阵X的MxN维度.我想为X的每一行创建一个对角矩阵.结果应为MxNxN.如何有效地做到这一点?谢谢!
Given the matrix X in dimension of MxN. I want to create a diagonal matrix for every row of X. The result should be in MxNxN. How to do it efficiently? Thank you!
推荐答案
out = np.zeros((m, n, n))
out[:, np.arange(n), np.arange(n)] = X
这篇关于沿numpy沿一个轴对角化2d矩阵的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文