API

cot:

assignment(W, C, eps[, seed])

This function sloves an additive approximation of the assignment problem between two discrete distributions and returns assignment plan and cost.

assignment_torch(W, C, eps, device[, seed])

This function sloves an additive approximation of the assignment problem between two discrete distributions and returns assignment plan and cost.

transport_torch(DA, SB, C, eps, device)

This function sloves the additive approximation of optimal transport problem between two discrete distributions and returns the transports plan, dual variables and total cost.

transport_lmr(DA, SB, C, eps)

This function sloves the additive approximation of optimal transport problem between two discrete distributions and returns the approximated cost based on the graph-based additive approximation algorithm [1]_ for OT.

ot_profile(DA, SB, C, eps[, p])

This function computes the approximated Optimal Transport profile (OT-Profile) [3]_ between two discrete distributions by leveraging the LMR algorithm [1]_.

rpw([DA, SB, dist, eps, k, p])

Computes the approximated Robust Partial p-Wasserstein (RPW) distance [4]_ between two discrete distributions.

Main cot functions