https://github.com/tensorly/tensorly
Tip revision: 7146ca6de4e9aafb344bfa9a035f5f0b640aabca authored by Jean Kossaifi on 27 February 2017, 14:39:12 UTC
DOC: Minor update
DOC: Minor update
Tip revision: 7146ca6
test_tucker.py
import numpy as np
from numpy.testing import assert_array_equal, assert_array_almost_equal
from ..base import unfold, tensor_to_vec
from ..tucker import tucker_to_tensor, tucker_to_unfolded, tucker_to_vec
from ..tenalg import kronecker
def test_tucker_to_tensor():
"""Test for tucker_to_tensor"""
X = np.array([[[1, 13],
[4, 16],
[7, 19],
[10, 22]],
[[2, 14],
[5, 17],
[8, 20],
[11, 23]],
[[3, 15],
[6, 18],
[9, 21],
[12, 24]]])
ranks = [2, 3, 4]
U = [np.arange(R * s).reshape((R, s)) for (R, s) in zip(ranks, X.shape)]
true_res = np.array([[[390, 1518, 2646, 3774],
[1310, 4966, 8622, 12278],
[2230, 8414, 14598, 20782]],
[[1524, 5892, 10260, 14628],
[5108, 19204, 33300, 47396],
[8692, 32516, 56340, 80164]]])
res = tucker_to_tensor(X, U)
assert_array_equal(true_res, res)
def test_tucker_to_unfolded():
"""Test for tucker_to_unfolded
Notes
-----
Assumes that tucker_to_tensor is properly tested
"""
G = np.random.random((4, 3, 5, 2))
ranks = [2, 2, 3, 4]
U = [np.random.random((ranks[i], G.shape[i])) for i in range(G.ndim)]
full_tensor = tucker_to_tensor(G, U)
for mode in range(G.ndim):
assert_array_almost_equal(tucker_to_unfolded(G, U, mode), unfold(full_tensor, mode))
assert_array_almost_equal(tucker_to_unfolded(G, U, mode),
U[mode].dot(unfold(G, mode).dot(kronecker(U, skip_matrix=mode).T)))
def test_tucker_to_vec():
"""Test for tucker_to_vec
Notes
-----
Assumes that tucker_to_tensor works correctly
"""
G = np.random.random((4, 3, 5, 2))
ranks = [2, 2, 3, 4]
U = [np.random.random((ranks[i], G.shape[i])) for i in range(G.ndim)]
vec = tensor_to_vec(tucker_to_tensor(G, U))
assert_array_almost_equal(tucker_to_vec(G, U), vec)
assert_array_almost_equal(tucker_to_vec(G, U), kronecker(U).dot(tensor_to_vec(G)))