Sparsemax in PyTorch


(Taha) #1

Is there any PyTorch implementation of Sparsemax [1]? I found the following implementation in Torch:
SparseMax in Torch

[1] Martins, Andre, and Ramon Astudillo. “From Softmax to Sparsemax: A sparse model of attention and multi-label classification.” In ICML 2016.


#2

This might be useful: