arXiv Open Access 2019

Implementation of batched Sinkhorn iterations for entropy-regularized Wasserstein loss

Thomas Viehmann

Lihat Sumber

Abstrak

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

Topik & Kata Kunci

stat.ML cs.LG

Penulis (1)

Thomas Viehmann

Format Sitasi

APA MLA BibTeX

Viehmann, T. (2019). Implementation of batched Sinkhorn iterations for entropy-regularized Wasserstein loss. https://arxiv.org/abs/1907.01729

Akses Cepat

Lihat di Sumber

Informasi Jurnal

Tahun Terbit: 2019
Bahasa: en
Sumber Database: arXiv
Akses: Open Access ✓