type

costs.Sinkhorn

class Sinkhorn:

Soft-OT renormalisation wrapper for the differentiable path.

Materialises the inner cost expression, runs ``n_iter`` Sinkhorn iterations in log space at temperature ``epsilon``, and returns the negative log transport plan as a new cost matrix. Useful as the cost feed into :class:`~unitrack.assignment.SoftAssignment`.

Members

method

__call__Run the inner producer and Sinkhorn-renormalise its cost.
__init__

property

epsilon
inner
n_iter

Source: unitrack/costs/combinators.py:180