ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.update_features
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]
.