ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.update_barycenter#

GWBarycenterProblem.update_barycenter(transports, a)[source]#

Update the barycenter cost matrix.

Uses the eq. 14 and 15 of [Peyré et al., 2016].

Parameters:

  • transports (Array) – Transport maps of shape [num_measures, bar_size, max_measure_size].

  • a (Array) – Barycenter weights of shape [bar_size,].

Return type:

Array

Returns:

Update cost matrix of shape [bar_size, bar_size].