[BUG FIX] Optimize top-k mask construction: prevent unsafe gradient flow and eliminate unnecessary memory allocations#184
Open
LoserCheems wants to merge 1 commit into
Open
[BUG FIX] Optimize top-k mask construction: prevent unsafe gradient flow and eliminate unnecessary memory allocations#184LoserCheems wants to merge 1 commit into
LoserCheems wants to merge 1 commit into
