ott.solvers.linear.sinkhorn.SinkhornOutput
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)- rtype
Array
set
(**kwargs)Return a copy of self, with potential overwrites.
set_cost
(ot_prob, lse_mode, use_danskin)- rtype
Return the entropic map estimator.
transport_cost_at_geom
(other_geom)Return bare transport cost of current solution at any geometry.
Attributes
- rtype
Array
- rtype
Array
- rtype
Return transport cost in dual form of current solution.
Alias for field number 2
Alias for field number 0
Alias for field number 1
- rtype
- rtype
- rtype
Transport matrix if it can be instantiated.
- rtype
Alias for field number 4
Return transport cost of current solution at geometry.
Alias for field number 3
- rtype
Tuple
[Array
,Array
]
Sum of transport matrix.