https://github.com/tensorly/tensorly
Revision 1bb217a077d6fa1d507f963a60da81adfd099d79 authored by Jean Kossaifi on 14 July 2017, 03:03:33 UTC, committed by GitHub on 14 July 2017, 03:03:33 UTC
Improving partial_svd by omitting full svd matrices when possible
Tip revision: 1bb217a077d6fa1d507f963a60da81adfd099d79 authored by Jean Kossaifi on 14 July 2017, 03:03:33 UTC
Merge pull request #7 from chubei/master
Merge pull request #7 from chubei/master
Tip revision: 1bb217a
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)))
Computing file changes ...