Implementation of batched Sinkhorn iterations for entropy-regularized Wasserstein loss
In this report, we review the calculation of entropy-regularised Wasserstein loss introduced by Cuturi and document a practical implementation in PyTorch. Code is available at this https URL