ott.core.sinkhorn_lr.LRSinkhornOutput#

class ott.core.sinkhorn_lr.LRSinkhornOutput(q=None, r=None, g=None, costs=None, reg_ot_cost=None, ot_prob=None)[source]#

Implement the problems.Transport interface, for a LR Sinkhorn solution.

Methods

apply(inputs[, axis])

Apply the transport to a ndarray; axis=1 for its transpose.

compute_reg_ot_cost(ot_prob[, use_danskin])

rtype

float

cost_at_geom(other_geom)

Return 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

ndarray

set(**kwargs)

Return a copy of self, with potential overwrites.

set_cost(ot_prob, lse_mode[, use_danskin])

rtype

LRSinkhornOutput

transport_mass()

Sum of transport matrix.

Attributes

a

rtype

ndarray

b

rtype

ndarray

converged

rtype

bool

costs

Alias for field number 3

g

Alias for field number 2

geom

rtype

Geometry

linear

rtype

bool

linear_output

rtype

bool

matrix

Transport matrix if it can be instantiated.

ot_prob

Alias for field number 5

q

Alias for field number 0

r

Alias for field number 1

reg_ot_cost

Alias for field number 4

Parameters