apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-12 05:22:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added faster einsums to linalg, with a couple of tests

Browse source
This commit is contained in:
Alan Saul 2015-04-30 11:16:29 +01:00
parent 5ad38ac640
commit 762e1e75b0
3 changed files with 35 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

17
GPy/testing/linalg_test.py
View file

@ -1,6 +1,7 @@
import numpy as np
import scipy as sp
from ..util.linalg import jitchol
from GPy.util.linalg import jitchol
import GPy
class LinalgTests(np.testing.TestCase):
def setUp(self):
@ -35,3 +36,17 @@ class LinalgTests(np.testing.TestCase):
return False
except sp.linalg.LinAlgError:
return True
def test_einsum_ijk_jlk_to_il(self):
A = np.random.randn(50, 150, 5)
B = np.random.randn(150, 100, 5)
pure = np.einsum('ijk,jlk->il', A, B)
quick = GPy.util.linalg.ijk_jlk_to_il(A, B)
np.testing.assert_allclose(pure, quick)
def test_einsum_ij_jlk_to_ilk(self):
A = np.random.randn(15, 150, 5)
B = np.random.randn(150, 50, 5)
pure = np.einsum('ijk,jlk->il', A, B)
quick = GPy.util.linalg.ijk_jlk_to_il(A,B)
np.testing.assert_allclose(pure, quick)