cb3c3798eabae57bab06227d0e77687c14bf099b,geomstats/stiefel.py,StiefelCanonicalMetric,inner_product,#StiefelCanonicalMetric#Any#Any#Any#,84

Before Change

        tangent_vec_b = gs.to_ndarray(tangent_vec_a, to_ndim=3)
        n_tangent_vecs_b, _, _ = tangent_vec_b.shape

        assert n_tangent_vecs_a == n_tangent_vecs_b

        inner_prod = gs.einsum("nij,nij->n", tangent_vec_a, tangent_vec_b)

        return inner_prod

After Change

        base_point_transpose = gs.transpose(base_point, axes=(0, 2, 1))

        aux = gs.matmul(
            gs.transpose(tangent_vec_a, axes=(0, 2, 1)),
            gs.eye(self.n) - 0.5 * gs.matmul(base_point, base_point_transpose))
        inner_prod = gs.trace(gs.matmul(aux, tangent_vec_b))
        return inner_prod

    def exp(self, tangent_vec, base_point):
        

Instances