ott.tools.unreg.HungarianOutput

class ott.tools.unreg.HungarianOutput(geom, paired_indices=None)

Output of the Hungarian solver.

Parameters:

• geom (Geometry) – geometry object

• paired_indices (Array | None) – Array of shape [2, n], of \(n\) pairs of indices, for which the optimal transport assigns mass. Namely, for each index \(0 \leq k < n\), if one has \(i := \text{paired_indices}[0, k]\) and \(j := \text{paired_indices}[1, k]\), then point \(i\) in the first geometry sends mass to point \(j\) in the second.

Methods

count(value, /)	Return number of occurrences of value.
index(value[, start, stop])	Return first index of value.

Attributes

geom	Alias for field number 0
matrix	[n, n] transport matrix in sparse format, with n NNZ entries.
paired_indices	Alias for field number 1