Python block_conjgradの例

コード例 #1

ファイル: test_solvers.py プロジェクト: falconlulu/nengo

def test_conjgrad(rng):
    A, b = get_system(1000, 100, 2, rng=rng)
    sigma = 0.1 * A.max()

    x0, _ = cholesky(A, b, sigma)
    x1, _ = conjgrad(A, b, sigma, tol=1e-3)
    x2, _ = block_conjgrad(A, b, sigma, tol=1e-3)
    assert np.allclose(x0, x1, atol=1e-6, rtol=1e-3)
    assert np.allclose(x0, x2, atol=1e-6, rtol=1e-3)

コード例 #2

def test_conjgrad(rng):
    A, b = get_system(1000, 100, 2, rng=rng)
    sigma = 0.1 * A.max()

    x0, _ = cholesky(A, b, sigma)
    x1, _ = conjgrad(A, b, sigma, tol=1e-3)
    x2, _ = block_conjgrad(A, b, sigma, tol=1e-3)
    assert np.allclose(x0, x1, atol=1e-6, rtol=1e-3)
    assert np.allclose(x0, x2, atol=1e-6, rtol=1e-3)