37d6a53cad2695ee387cfa6ffd0e88f96e8ff699,geomstats/special_orthogonal_group.py,SpecialOrthogonalGroup,belongs,#SpecialOrthogonalGroup#Any#,96

Before Change

        Check that a vector belongs to the
        special orthogonal group.
        
        if rot_vec.ndim == 1:
            rot_vec = np.expand_dims(rot_vec, axis=0)

        assert rot_vec.ndim == 2

        return rot_vec.shape[1] == self.dimension

    def regularize(self, rot_vec):
        

After Change

        Check that a vector belongs to the
        special orthogonal group.
        
        rot_vec = vectorization_utils.expand_dims(rot_vec, to_dim=2)
        _, vec_dim = rot_vec.shape
        return vec_dim == self.dimension

    def regularize(self, rot_vec):
        

Instances