ac24bfe744774df11dfc075397970c75e0fc5206,geomstats/geometry/lie_algebra.py,MatrixLieAlgebra,orthonormal_basis,#MatrixLieAlgebra#,166

Before Change

        basis : array-like, shape=[dim, n, n]
            Orthonormal basis.
        
        metric_matrix = self.reshape_metric_matrix(metric_matrix) + gs.eye(
            self.n)
        return self.basis / gs.sqrt(2 * metric_matrix)

    def projection(self, mat):
        Project a matrix to the Lie Algebra.

After Change

        
        metric_matrix = self.reshape_metric_matrix(metric_matrix)
        norms = gs.sum(
            metric_matrix * self.basis * self.basis, (-2, -1))

        return gs.einsum("i, ikl->ikl", 1. / gs.sqrt(norms), self.basis)

Instances