def test_project(self):
def random_tanget_space_point(X):
coresX = tt.tensor.to_list(X)
point = 0 * tt.ones(X.n)
for dim in range(X.d):
curr = deepcopy(coresX)
curr[dim] = np.random.rand(curr[dim].shape[0],
curr[dim].shape[1],
curr[dim].shape[2])
point += tt.tensor.from_list(curr)
return point
for debug_mode in [False, True]:
for use_jit in [False, True]:
X = tt.rand([4, 4, 4], 3, [1, 4, 4, 1])
Z = random_tanget_space_point(X)
PZ = riemannian.project(X, Z, use_jit=use_jit,
debug=debug_mode)
np.testing.assert_array_almost_equal(Z.full(), PZ.full())
X = tt.rand([2, 3, 4], 3, [1, 5, 4, 1])
Z = random_tanget_space_point(X)
PZ = riemannian.project(X, Z, use_jit=use_jit,
debug=debug_mode)
np.testing.assert_array_almost_equal(Z.full(), PZ.full())
def test_project_sum_equal_ranks(self):
for debug_mode in [False, True]:
for use_jit in [False, True]:
X = tt.rand([4, 4, 4], 3, [1, 4, 4, 1])
Z = [0] * 7
for idx in range(7):
Z[idx] = tt.rand([4, 4, 4], 3, [1, 2, 3, 1])
project_sum = riemannian.project(X, Z, use_jit=use_jit, debug=debug_mode)
sum_project = X * 0
for idx in range(len(Z)):
sum_project += riemannian.project(X, Z[idx], use_jit=use_jit, debug=debug_mode)
np.testing.assert_array_almost_equal(sum_project.full(), project_sum.full())
def test_project_sum_equal_ranks(self):
for debug_mode in [False, True]:
for use_jit in [False, True]:
X = tt.rand([4, 4, 4], 3, [1, 4, 4, 1])
Z = [0] * 7
for idx in range(7):
Z[idx] = tt.rand([4, 4, 4], 3, [1, 2, 3, 1])
project_sum = riemannian.project(X, Z, use_jit=use_jit,
debug=debug_mode)
sum_project = X * 0
for idx in range(len(Z)):
sum_project += riemannian.project(X, Z[idx],
use_jit=use_jit,
debug=debug_mode)
np.testing.assert_array_almost_equal(sum_project.full(),
project_sum.full())
def test_project(self):
def random_tanget_space_point(X):
coresX = tt.tensor.to_list(X)
point = 0 * tt.ones(X.n)
for dim in range(X.d):
curr = deepcopy(coresX)
curr[dim] = np.random.rand(curr[dim].shape[0], curr[dim].shape[1], curr[dim].shape[2])
point += tt.tensor.from_list(curr)
return point
for debug_mode in [False, True]:
for use_jit in [False, True]:
X = tt.rand([4, 4, 4], 3, [1, 4, 4, 1])
Z = random_tanget_space_point(X)
PZ = riemannian.project(X, Z, use_jit=use_jit, debug=debug_mode)
np.testing.assert_array_almost_equal(Z.full(), PZ.full())
X = tt.rand([2, 3, 4], 3, [1, 5, 4, 1])
Z = random_tanget_space_point(X)
PZ = riemannian.project(X, Z, use_jit=use_jit, debug=debug_mode)
np.testing.assert_array_almost_equal(Z.full(), PZ.full())
