ott.problems.linear.barycenter_problem.FixedBarycenterProblem

ott.problems.linear.barycenter_problem.FixedBarycenterProblem#

class ott.problems.linear.barycenter_problem.FixedBarycenterProblem(geom, a, weights=None)[source]#

Fixed-support Wasserstein barycenter problem.

Parameters:
  • geom (Geometry) – Geometry object.

  • a (Array) – batch of histograms of shape [batch, num_a] where num_a matches the first value of the shape attribute of geom.

  • weights (Optional[Array]) – [batch,] positive weights summing to \(1\). Uniform by default.

Methods

Attributes

num_measures

Number of measures.

weights

math`1`.