Sinkhorn layer in keras, inspired by the reference implementation of numbers sorting task solution (please see the 'Learning Latent Permutations with Gumbel-Sinkhorn Networks' paper by Gonzalo Mena, David Belanger, Scott Linderman, Jasper Snoek).
Requirements:
- Python3
- keras, scipy, numpy
Bug reports, comments and enhancements are welcome.
Not neccessary, but is greatly appreciated.
@misc{Alekseev2018sinkhornkeras,
author = {Alekseev~A.M.},
title = {Sinkhorn layer, Keras implementation.},
year = {2018},
publisher = {GitHub},
journal = {GitHub repository},
howpublished = {\url{https://github.com/alexeyev/SinkhornLayer-keras}},
commit = {the latest commit of the codebase you have used}
}