120b9f6d5bb7c0c7dde5fb60cdb503dddaeb03a0,geomstats/geometry/special_euclidean.py,_SpecialEuclideanVectors,matrix_from_vector,#_SpecialEuclideanVectors#Any#,305

Before Change

        trans_vec = vec[:, self.rotations.dim:]

        rot_mat = self.rotations.matrix_from_rotation_vector(rot_vec)
        trans_vec = gs.reshape(trans_vec, (n_vecs, self.n, 1))
        mat = gs.concatenate((rot_mat, trans_vec), axis=2)
        last_lines = gs.array(gs.get_mask_i_float(self.n, self.n + 1))
        last_lines = gs.to_ndarray(last_lines, to_ndim=2)

After Change

        trans_vec = vec[..., self.rotations.dim:]

        rot_mat = self.rotations.matrix_from_rotation_vector(rot_vec)
        return embed(rot_mat, trans_vec, output_shape)

    @geomstats.vectorization.decorator(
        ["else", "vector", "vector"])
    def compose(self, point_a, point_b):

Instances