Implementation of batched Sinkhorn iterations for entropy-regularized Wasserstein loss

Open in new window