Implementation of batched Sinkhorn iterations for entropy-regularized Wasserstein loss
2019-07-01Code Available0· sign in to hype
Thomas Viehmann
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/t-vi/pytorch-tvmiscOfficialIn paperpytorch★ 477
- github.com/zchenCN/OptimalTransportnone★ 0
Abstract
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 https://github.com/t-vi/pytorch-tvmisc/blob/master/wasserstein-distance/Pytorch_Wasserstein.ipynb