			#!/usr/bin/env python3
			# -- coding: utf-8 --
			"""
			Created on Mon Oct 15 14:29:42 2018

			@author: sp4e
			"""

			import numpy as np
			from scipy.sparse.linalg import cg
			import time
			import math
			import numpy


			A=np.array([[4,1],[1,3]])
			b=np.array([[1],[2]])
			x=np.array([[0],[0]])
			def conjugate_grad(A, b, x):
			n = len(b)
			r = b-np.dot(A, x)
			p = r
			rsold = np.dot(numpy.transpose(r), r)
			for i in range (1,10):
			Ap = np.dot(A, p)
			alpha = rsold / np.dot(numpy.transpose(p), Ap)
			x = x+alpha * p
			r = r-alpha * Ap
			rsnew = np.dot(numpy.transpose(r), r)
			beta = rsnew / rsold
			rsold= rsnew
			if abs(rsnew) < 1e-10:
			print (i)
			break
			p = beta * p + r
			return x


			print (A);
			print (b);
			print (x);
			y=conjugate_grad(A, b, x)
			print(y)

SP4EHW/hw2.py
53998921e32bmaster

hw2.py
View Options