def test_unfolding_size(self):
"""Test size of all the unfoldings of a tensor.
"""
A = np.zeros((2, 3, 4))
A0 = tensorial_kernel.unfold(A, mode=0)
assert (A0.shape == (2, 12))
A1 = tensorial_kernel.unfold(A, mode=1)
assert (A1.shape == (3, 8))
A2 = tensorial_kernel.unfold(A, mode=2)
assert (A2.shape == (4, 6))
def test_unfolding_size(self):
"""Test size of all the unfoldings of a tensor.
"""
A = np.zeros((2,3,4))
A0 = tensorial_kernel.unfold(A, mode=0)
assert(A0.shape==(2,12))
A1 = tensorial_kernel.unfold(A, mode=1)
assert(A1.shape==(3,8))
A2 = tensorial_kernel.unfold(A, mode=2)
assert(A2.shape==(4,6))
def test_unfolding_order(self):
# A = np.arange(np.prod((2,3,4))).reshape(2,3,4)
A = np.array([[[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]],
[[12, 13, 14, 15], [16, 17, 18, 19], [20, 21, 22, 23]]])
A0 = tensorial_kernel.unfold(A, mode=0)
A0_expected = np.array(
[[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11],
[12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23]])
assert ((A0 == A0_expected).all())
A1 = tensorial_kernel.unfold(A, mode=1)
A1_expected = np.array([[0, 1, 2, 3, 12, 13, 14, 15],
[4, 5, 6, 7, 16, 17, 18, 19],
[8, 9, 10, 11, 20, 21, 22, 23]])
assert ((A1 == A1_expected).all())
A2 = tensorial_kernel.unfold(A, mode=2)
A2_expected = np.array([[0, 12, 4, 16, 8, 20], [1, 13, 5, 17, 9, 21],
[2, 14, 6, 18, 10, 22], [3, 15, 7, 19, 11,
23]])
assert ((A2 == A2_expected).all())
def test_unfolding_order(self):
# A = np.arange(np.prod((2,3,4))).reshape(2,3,4)
A = np.array([[[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]],
[[12, 13, 14, 15],
[16, 17, 18, 19],
[20, 21, 22, 23]]])
A0 = tensorial_kernel.unfold(A, mode=0)
A0_expected = np.array([[ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11],
[12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23]])
assert((A0 == A0_expected).all())
A1 = tensorial_kernel.unfold(A, mode=1)
A1_expected = np.array([[ 0, 1, 2, 3, 12, 13, 14, 15],
[ 4, 5, 6, 7, 16, 17, 18, 19],
[ 8, 9, 10, 11, 20, 21, 22, 23]])
assert((A1 == A1_expected).all())
A2 = tensorial_kernel.unfold(A, mode=2)
A2_expected = np.array([[ 0, 12, 4, 16, 8, 20],
[ 1, 13, 5, 17, 9, 21],
[ 2, 14, 6, 18, 10, 22],
[ 3, 15, 7, 19, 11, 23]])
assert((A2 == A2_expected).all())