Method __init__

intermediate_source/parametrizations.py:224–226

(self, n)

# orthogonal weights
class CayleyMap(nn.Module):
    def __init__(self, n):
        super().__init__()
        self.register_buffer("Id", torch.eye(n))

    def forward(self, X):
        # (I + X)(I - X)^{-1}