ott.problems.quadratic.gw_barycenter.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 SqEuclidean cost.

Parameters

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

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

Return type

Optional[Array]

Returns

Updated features of shape [bar_size, ndim_fused].