difftopk icon indicating copy to clipboard operation
difftopk copied to clipboard

About difftopk and torch.topk

Open lijun2005 opened this issue 9 months ago • 1 comments

Great job! This is the first time I've seen a differentiable top-k selection in an engineering project so well-implemented. But I have a question for the author: What are the differences between the top-k implementation in this project and the one built into PyTorch (torch.topk )?

lijun2005 avatar May 06 '24 10:05 lijun2005