gpt4 book ai didi

python - 计算大型稀疏矩阵的迹

转载 作者:太空宇宙 更新时间:2023-11-04 12:40:34 24 4
gpt4 key购买 nike

我有一个稀疏大矩阵(线性维度为 2*10^6),我想计算它的轨迹。计算它的蛮力需要 16 秒来访问每个对角线元素(因此我可以在一年内完成!)。

我正在考虑使用 scipy.io.mmwrite 将它保存到磁盘并使用应该更快的 c++ 代码读取它。但是我找不到任何可以帮助我做到这一点的软件包。

如有任何建议,我们将不胜感激!

谢谢。

最佳答案

解决方案很简单:np.array(Mat.diagonal()).sum()

谢谢@hpaulj!

关于python - 计算大型稀疏矩阵的迹,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54379162/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com