Exemple #1
0
def test_right_rotate_trace():
    a = torch.randn(5, 5)

    for i in range(1, 5):
        actual = circulant.rotate(a, -i).trace()
        result = circulant.left_rotate_trace(a, -i)

        assert(math.fabs(actual - result) < 1e-5)
Exemple #2
0
    def test_left_rotate_trace(self):
        a = torch.randn(5, 5)

        for i in range(1, 5):
            actual = circulant.rotate(a, i).trace()
            result = circulant.left_rotate_trace(a, i)

            self.assertLess(math.fabs(actual - result), 1e-5)