successive-halving-topk
successive-halving-topk copied to clipboard
A fast and highly accurate differentiable Top-k operator from the "Successive Halving Top-k Operator" AAAI'21 paper.
Results
1
successive-halving-topk issues
Sort by
recently updated
recently updated
newest added
The indicators obtained by pytorch's topk operation are not differentiable. How can I use the topk here to obtain differentiable indicators? My application cannot tolerate changes in embeddings. So I...