inner_prod : array-like, shape=[...]
Inner-product of the two tangent vectors.
aux_prod = tangent_vec_a * tangent_vec_b
metric_mat = self.metric_mat_at_identity
if (Matrices.is_diagonal(metric_mat)
and self.lie_algebra is not None):
aux_prod *= self.reshaped_metric_matrix
inner_prod = gs.sum(aux_prod, axis=(-2, -1))
return inner_prod
def inner_product(self, tangent_vec_a, tangent_vec_b, base_point=None):
Compute inner product of two vectors in tangent space at base point.
After Change
if (Matrices.is_diagonal(metric_mat)
and self.lie_algebra is not None):
tan_b *= self.reshaped_metric_matrix
inner_prod = Matrices.frobenius_product(
tangent_vec_a, tan_b)
return inner_prod
def inner_product(self, tangent_vec_a, tangent_vec_b, base_point=None):
Compute inner product of two vectors in tangent space at base point.
