ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput#
- class ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput(q, r, g, costs, errors, ot_prob, reg_ot_cost=None)[source]#
Implement the problems.Transport interface, for a LR Sinkhorn solution.
Methods
apply
(inputs[, axis])Apply the transport to a array; axis=1 for its transpose.
compute_reg_ot_cost
(ot_prob[, use_danskin])- rtype
cost_at_geom
(other_geom)Return OT cost for current solution, evaluated at any 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
transport_cost_at_geom
(other_geom)Return (by recomputing it) bare transport cost of current solution.
Attributes
- rtype
Array
- rtype
Array
- rtype
Alias for field number 3
Alias for field number 4
Alias for field number 2
- rtype
- rtype
- rtype
Transport matrix if it can be instantiated.
Alias for field number 5
Return (by recomputing it) transport cost of current solution.
Alias for field number 0
Alias for field number 1
Alias for field number 6
Sum of transport matrix.
- Parameters
q (jax.Array) –
r (jax.Array) –
g (jax.Array) –
costs (jax.Array) –
errors (jax.Array) –
ot_prob (ott.problems.linear.linear_problem.LinearProblem) –