Tutorials:
Benchmarks:
Advanced Applications:
Public API: ott packages
References:
Carry out one Sinkhorn update for scalings, using kernel directly.
scaling (ndarray) – jnp.ndarray of num_a or num_b positive values.
ndarray
marginal (ndarray) – targeted marginal
iteration (Optional[int]) – used to compute epsilon from schedule, if provided.
Optional
int
axis (int) – axis along which the update should be carried out.
new scaling vector, of size num_b if axis=0, num_a if axis is 1.