Implementation of Sparsemax activation in Pytorch - View it on GitHub