ott.solvers.linear.sinkhorn.SinkhornOutput#
- class ott.solvers.linear.sinkhorn.SinkhornOutput(f=None, g=None, errors=None, reg_ot_cost=None, ot_prob=None)[source]#
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:
Return the entropic map estimator.
transport_cost_at_geom
(other_geom)Return bare transport cost of current solution at any geometry.
Attributes
Return transport cost in dual form of current solution.
Alias for field number 2
Alias for field number 0
Alias for field number 1
Transport matrix if it can be instantiated.
Alias for field number 4
Return transport cost of current solution at geometry.
Alias for field number 3
Sum of transport matrix.