Fixed bugs with givens-rotator for eigen

2022-10-24 10:08:31 +02:00
parent e31c25135e
commit 82a174122a
2 changed files with 8 additions and 12 deletions
@@ -59,8 +59,3 @@ class Rotation(nn.Module):
        for idx, (i, j) in reversed(list(enumerate(itertools.combinations(range(self.D), 2)))):
            x = torch.matmul(x, G_transpose(self.D, i, j, -self.theta[idx]))
        return x
-
-
-if __name__ == '__main__':
-    import doctest
-    doctest.testmod()