2d199a2b947146cb788b6c81feb83a1f6bd196b2,geomstats/special_orthogonal_group.py,SpecialOrthogonalGroup,regularize_tangent_vec_at_identity,#SpecialOrthogonalGroup#Any#Any#Any#,167

Before Change

                mask_else = ~mask_0

                mask_0 = gs.squeeze(mask_0, axis=1)
                mask_else = gs.squeeze(mask_else, axis=1)

                coef = gs.empty_like(tangent_vec_metric_norm)
                regularized_vec = tangent_vec

                regularized_vec[mask_0] = tangent_vec[mask_0]

                coef[mask_else] = (tangent_vec_metric_norm[mask_else]
                                   / tangent_vec_canonical_norm[mask_else])
                regularized_vec[mask_else] = self.regularize(
                        coef[mask_else] * tangent_vec[mask_else])
                regularized_vec[mask_else] = (regularized_vec[mask_else]

After Change

                coef += mask_else_float * (
                        tangent_vec_metric_norm
                        / tangent_vec_canonical_norm)
                regularized_vec += mask_else_float * self.regularize(
                        coef * tangent_vec)
                // This avois dividing by 0.
                coef += mask_0_float * 1.
                regularized_vec = mask_else_float * (
                        regularized_vec / coef)

Instances