successive-halving-topk icon indicating copy to clipboard operation
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...