ott.core.bar_problems.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 (ndarray) – Transport maps of shape [num_measures, bar_size, max_measure_size].

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

Return type

ndarray

Returns

Update cost matrix of shape [bar_size, bar_size].