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