实现共轭梯度法(Conjugate Gradient Method)来求解大型稀疏对称正定线性方程组。共轭梯度法是一种高效的迭代算法,用于求解形如 的线性方程组。 请不要使用scipy的cg函数。
输入描述:
第一行输入一个二维数组A,表示系数矩阵。第二行输入一个一维数组b,表示等式右侧向量。第三行输入一个整数n,表示最大迭代次数。


输出描述:
返回一个一维numpy数组,表示方程组的解向量 。
示例1

输入

[[4, 1], [1, 3]]
[1, 2]
100

输出

[0.09090909 0.63636364]

备注:
1.对应的输入、输出已给出,您只用实现核心功能函数即可。2.支持numpy、scipy、pandas、scikit-learn库。
加载中...