ott.core.sinkhorn_lr.LRSinkhorn.recompute_couplings

LRSinkhorn.recompute_couplings(f1, g1, c_q, f2, g2, c_r, h)

Return type

Tuple[ndarray, ndarray, ndarray]

Parameters

• f1 (jax._src.numpy.ndarray.ndarray) –

• g1 (jax._src.numpy.ndarray.ndarray) –

• c_q (jax._src.numpy.ndarray.ndarray) –

• f2 (jax._src.numpy.ndarray.ndarray) –

• g2 (jax._src.numpy.ndarray.ndarray) –

• c_r (jax._src.numpy.ndarray.ndarray) –

• h (jax._src.numpy.ndarray.ndarray) –