33a866645dd50158129744018266d0837ef94392,geomstats/special_orthogonal_group.py,SpecialOrthogonalGroup,skew_matrix_from_vector,#SpecialOrthogonalGroup#Any#,306

Before Change

        vec = gs.to_ndarray(vec, to_ndim=2)
        n_vecs, vec_dim = vec.shape

        vec_dim = gs.cast(vec_dim, gs.float32)
        mat_dim = int((1. + gs.sqrt(1. + 8. * vec_dim)) / 2.)
        assert mat_dim == self.n

        skew_mat = gs.zeros((n_vecs,) + (self.n,) * 2)

After Change

        vec = gs.to_ndarray(vec, to_ndim=2)
        n_vecs, vec_dim = vec.shape

        vec_dim = float(vec_dim)  //gs.cast(vec_dim, gs.float32)
        mat_dim = int((1. + gs.sqrt(1. + 8. * vec_dim)) / 2.)
        assert mat_dim == self.n

Instances