79ab2da38070ed0a972e72c16a44bd266ed6e941,geomstats/geometry/special_orthogonal.py,SpecialOrthogonal,regularize_tangent_vec_at_identity,#SpecialOrthogonal#Any#Any#Any#,186

Before Change

                coef += mask_else_float * (
                    tangent_vec_metric_norm
                    / tangent_vec_canonical_norm)
                regularized_vec += mask_else_float * self.regularize(
                    coef * tangent_vec)
                coef += mask_0_float
                regularized_vec = mask_else_float * (
                    regularized_vec / coef)
            else:
                // TODO(nina): Check if/how regularization is needed in nD?
                regularized_vec = tangent_vec

After Change

                regularized_vec += gs.einsum(
                    "...,...i->...i",
                    mask_else_float,
                    self.regularize(coef_tangent_vec))

                coef += mask_0_float
                regularized_vec = gs.einsum(
                    "...,...i->...i", 1. / coef, regularized_vec)
                regularized_vec = gs.einsum(
                    "...,...i->...i", mask_else_float, regularized_vec)
            else:
                // TODO(nina): Check if/how regularization is needed in nD?

Instances