EMD: Earth Mover Distance It contains the Earth Mover Distance function with CUDA implementation for PyTorch as well as Tensorflow. Test Entire Package: chmod +x test.sh ./test.sh Some Important Points about usage: Both Inputs should have dimension (Batch x Number of Points x Dimension) ...
CudaOT Repository for solving discrete optimal transport problems via Cplex, Sinkhorn, FastEMD and the GPU implementation of the Multi-Scale Sparse Sinkhorn. This library is designed to solve a very large instance of discrete optimal transport problems. ...