ott.solvers.linear.sinkhorn.SinkhornOutput

class ott.solvers.linear.sinkhorn.SinkhornOutput(f=None, g=None, errors=None, reg_ot_cost=None, ot_prob=None)

Implements the problems.Transport interface, for a Sinkhorn solution.

Methods

apply(inputs[, axis])	Apply the transport to a ndarray; axis=1 for its transpose.
cost_at_geom(other_geom)	Return reg-OT cost for matrix, evaluated at other cost matrix.
count(value, /)	Return number of occurrences of value.
index(value[, start, stop])	Return first index of value.
marginal(axis)	param axis:
set(**kwargs)	Return a copy of self, with potential overwrites.
set_cost(ot_prob, lse_mode, use_danskin)	param ot_prob:
to_dual_potentials()	Return the entropic map estimator.
transport_cost_at_geom(other_geom)	Return bare transport cost of current solution at any geometry.

Attributes

a
b
converged
dual_cost	Return transport cost in dual form of current solution.
errors	Alias for field number 2
f	Alias for field number 0
g	Alias for field number 1
geom
linear
linear_output
matrix	Transport matrix if it can be instantiated.
n_iters
ot_prob	Alias for field number 4
primal_cost	Return transport cost of current solution at geometry.
reg_ot_cost	Alias for field number 3
scalings
transport_mass	Sum of transport matrix.

Parameters:

• f (Optional[Array]) –

• g (Optional[Array]) –

• errors (Optional[Array]) –

• reg_ot_cost (Optional[float]) –

• ot_prob (Optional[LinearProblem]) –