03eb1fad7a8e3accc597222625b3bfabaaf1486a,geomstats/discretized_curves_space.py,SRVMetric,exp,#SRVMetric#Any#Any#,283

Before Change

            self.pointwise_inner_product(tangent_vec_derivative,
                                         base_curve_velocity,
                                         base_curve[:, :-1, :])
        srv_initial_derivative = gs.transpose(coef_1) * \
            gs.transpose(tangent_vec_derivative, (2, 1, 0)) + \
            gs.transpose(coef_2) * gs.transpose(base_curve_velocity, (2, 1, 0))
        srv_initial_derivative = gs.transpose(srv_initial_derivative,
                                              (2, 1, 0))

        end_curve_srv = self.l2_metric.exp(tangent_vec=srv_initial_derivative,
                                           base_curve=base_curve_srv)
        end_curve_starting_point = self.embedding_metric.exp(

After Change

        base_curve_velocity_norm = self.pointwise_norm(
                base_curve_velocity, base_curve[:, :-1, :])

        inner_prod = self.pointwise_inner_product(tangent_vec_derivative,
                                                  base_curve_velocity,
                                                  base_curve[:, :-1, :])
        coef_1 = 1 / gs.sqrt(base_curve_velocity_norm)
        coef_2 = - 1 / (2 * base_curve_velocity_norm ** (5/2)) * inner_prod

        term_1 = gs.einsum("ij,ijk->ijk", coef_1, tangent_vec_derivative)
        term_2 = gs.einsum("ij,ijk->ijk", coef_2, base_curve_velocity)
        srv_initial_derivative = term_1 + term_2

        end_curve_srv = self.l2_metric.exp(tangent_vec=srv_initial_derivative,
                                           base_curve=base_curve_srv)
        end_curve_starting_point = self.embedding_metric.exp(

Instances