TopK

  • 詳解torch.topk函數

    一、torch.topk函數 在深度學習領域中,我們通常需要對張量進行排序(如特徵選擇、模型解釋等),而PyTorch中的torch.topk()函數則是我們在進行此類操作時候的一…

    編程 2024-12-26
  • TopK演算法詳解

    一、TopK演算法例題 TopK演算法是一種尋找數據中前K大或前K小的演算法,在日常開發中經常會遇到需要尋找數據中前K大或前K小的情況,例如尋找銷售額前十的商品或者尋找某個數據集中最熱門…

    編程 2024-11-16