ott.core.bar_problems.GWBarycenterProblem.update_features#

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

Update the barycenter features in the fused case [Titouan et al., 2019].

Uses [Cuturi and Doucet, 2014] eq. 8, and is implemented only for the squared Euclidean cost.

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

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

Return type

Optional[ndarray]

Returns

Updated features of shape [bar_size, ndim_fused].