Implementation of batched Sinkhorn iterations for entropy-regularized Wasserstein loss

07/01/2019

∙

by   Thomas Viehmann, et al.

∙

0

∙

share

In this report, we review the calculation of entropy-regularised Wasserstein loss introduced by Cuturi and document a practical implementation in PyTorch.