ott.solvers.linear.sinkhorn.SinkhornOutput.transport_cost_at_geom

ott.solvers.linear.sinkhorn.SinkhornOutput.transport_cost_at_geom#

SinkhornOutput.transport_cost_at_geom(other_geom)[source]#

Return bare transport cost of current solution at any geometry.

In order to compute cost, we check first if the geometry can be converted to a low-rank cost geometry in order to speed up computations, without having to materialize the full cost matrix. If this is not possible, we resort to instantiating both transport matrix and cost matrix.

Parameters:

other_geom (Geometry) – geometry whose cost matrix is used to evaluate the transport cost.

Return type:

Array

Returns:

the transportation cost at \(C\), i.e. \(\langle P, C \rangle\).