2ffc1c8714980948b42c088880caa64bcc373a5e,geomstats/geometry/special_orthogonal.py,_SpecialOrthogonal2Vectors,matrix_from_rotation_vector,#_SpecialOrthogonal2Vectors#Any#,488

Before Change

        cos_matrix = cos_term * gs.array([gs.eye(2)] * n_samples)
        sin_term = gs.to_ndarray(gs.sin(rot_vec), to_ndim=3, axis=2)
        sin_matrix = -sin_term * self.skew_matrix_from_vector(
            gs.array([[1]] * n_samples))
        return cos_matrix + sin_matrix

    def compose(self, point_a, point_b):

After Change

        rot_vec = self.regularize(rot_vec)

        cos_term = gs.cos(rot_vec)
        cos_matrix = gs.einsum("...l,ij->...ij", cos_term, gs.eye(2))
        sin_term = gs.sin(rot_vec)
        sin_matrix = self.skew_matrix_from_vector(-sin_term)
        return cos_matrix + sin_matrix

Instances